WXM2000 commited on
Commit
d2fc8c9
·
1 Parent(s): 7b9c2ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeline
5
- model = AutoModelForSequenceClassification.from_pretrained('uer/roberta-base-finetuned-jd-binary-chinese',local_files_only=True)
6
- tokenizer = AutoTokenizer.from_pretrained('uer/roberta-base-finetuned-jd-binary-chinese',local_files_only=True)
7
  sentiment_classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
8
  examples=["小红正在吃一块美味的蛋糕。","小红在蛋糕里发现了一只苍蝇。"]
9
 
 
2
  from transformers import pipeline
3
 
4
  from transformers import AutoModelForSequenceClassification,AutoTokenizer,pipeline
5
+ model = AutoModelForSequenceClassification.from_pretrained('uer/roberta-base-finetuned-jd-binary-chinese')
6
+ tokenizer = AutoTokenizer.from_pretrained('uer/roberta-base-finetuned-jd-binary-chinese')
7
  sentiment_classifier = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
8
  examples=["小红正在吃一块美味的蛋糕。","小红在蛋糕里发现了一只苍蝇。"]
9