Spaces:

dahongj
/

sentiment-analysis

Runtime error

dahongj commited on Apr 9, 2023

Commit

746f5bc

•

1 Parent(s): af81e70

edit

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 def bertweet(data):
-    specific_model = AutoModelForSequenceClassification.from_pretrained(model="finiteautomata/bertweet-base-sentiment-analysis")
     result = specific_model(data)
     label = result[0]['label']
     score = result[0]['score']

 import torch
 def bertweet(data):
+    specific_model = AutoModelForSequenceClassification.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
     result = specific_model(data)
     label = result[0]['label']
     score = result[0]['score']