820nam commited on
Commit
ebdab57
ยท
verified ยท
1 Parent(s): 748540f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassifica
6
  # ๋”ฅ๋Ÿฌ๋‹ ๋ชจ๋ธ ๋กœ๋“œ
7
  @st.cache_resource
8
  def load_model():
9
- model_name = "nlptown/bert-base-multilingual-uncased-sentiment"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
12
  return pipeline("text-classification", model=model, tokenizer=tokenizer)
 
6
  # ๋”ฅ๋Ÿฌ๋‹ ๋ชจ๋ธ ๋กœ๋“œ
7
  @st.cache_resource
8
  def load_model():
9
+ model_name = "bucketresearch/politicalBiasBERT" # ์ •์น˜ ์„ฑํ–ฅ ๋ถ„์„์— ํŠนํ™”๋œ ๋ชจ๋ธ
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
  model = AutoModelForSequenceClassification.from_pretrained(model_name)
12
  return pipeline("text-classification", model=model, tokenizer=tokenizer)