shrish191 commited on
Commit
01d25c9
·
verified ·
1 Parent(s): 2f53d0e

Update evaluate.py

Browse files
Files changed (1) hide show
  1. 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
- fallback_tokenizer = AutoTokenizer.from_pretrained(fallback_model_name)
15
- fallback_model = AutoModelForSequenceClassification.from_pretrained(fallback_model_name)
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")