blockenters commited on
Commit
bbe6e89
ยท
1 Parent(s): 4aaae78
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,9 +1,11 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
 
4
  # 1. ๊ฐ์ • ๋ถ„์„ ๋ชจ๋ธ ๋กœ๋“œ
5
- model_name = "jaehyeong/koelectra-small-finetuned-nsmc"
6
- sentiment_analyzer = pipeline("sentiment-analysis", model=model_name, tokenizer=model_name)
 
 
7
 
8
  # 2. ๊ฐ์ • ๋ถ„์„ ํ•จ์ˆ˜ ์ •์˜
9
  def analyze_sentiment_korean(text):
 
1
  import gradio as gr
2
+ from transformers import pipeline, AutoModelForSequenceClassification, AutoTokenizer
3
 
4
  # 1. ๊ฐ์ • ๋ถ„์„ ๋ชจ๋ธ ๋กœ๋“œ
5
+ model_name = "kykim/bert-kor-base"
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModelForSequenceClassification.from_pretrained(model_name)
8
+ sentiment_analyzer = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
9
 
10
  # 2. ๊ฐ์ • ๋ถ„์„ ํ•จ์ˆ˜ ์ •์˜
11
  def analyze_sentiment_korean(text):