Spaces:
Sleeping
Sleeping
Update evaluate.py
Browse files- evaluate.py +2 -2
evaluate.py
CHANGED
|
@@ -11,8 +11,8 @@ def get_classification_report():
|
|
| 11 |
#tokenizer = AutoTokenizer.from_pretrained("Shrish/mbert-sentiment")
|
| 12 |
#model = TFAutoModelForSequenceClassification.from_pretrained("Shrish/mbert-sentiment")
|
| 13 |
fallback_model_name = "cardiffnlp/twitter-roberta-base-sentiment"
|
| 14 |
-
|
| 15 |
-
|
| 16 |
|
| 17 |
# Tokenize and predict
|
| 18 |
inputs = tokenizer(texts, padding=True, truncation=True, return_tensors="tf")
|
|
|
|
| 11 |
#tokenizer = AutoTokenizer.from_pretrained("Shrish/mbert-sentiment")
|
| 12 |
#model = TFAutoModelForSequenceClassification.from_pretrained("Shrish/mbert-sentiment")
|
| 13 |
fallback_model_name = "cardiffnlp/twitter-roberta-base-sentiment"
|
| 14 |
+
tokenizer = AutoTokenizer.from_pretrained(fallback_model_name)
|
| 15 |
+
model = AutoModelForSequenceClassification.from_pretrained(fallback_model_name)
|
| 16 |
|
| 17 |
# Tokenize and predict
|
| 18 |
inputs = tokenizer(texts, padding=True, truncation=True, return_tensors="tf")
|