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")
|