shrish191 commited on
Commit
cbf8340
·
verified ·
1 Parent(s): ae4aef6

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
  true_labels = df["label"].tolist()
12
 
13
  # Load tokenizer and model
14
- tokenizer = AutoTokenizer.from_pretrained("Shrish/mbert-sentiment")
15
- model = TFAutoModelForSequenceClassification.from_pretrained("Shrish/mbert-sentiment")
16
 
17
  # Tokenize
18
  inputs = tokenizer(texts, padding=True, truncation=True, return_tensors="tf")
 
11
  true_labels = df["label"].tolist()
12
 
13
  # Load tokenizer and model
14
+ tokenizer = AutoTokenizer.from_pretrained("shrish191/sentiment-bert")
15
+ model = TFAutoModelForSequenceClassification.from_pretrained("shrish191/sentiment-bert")
16
 
17
  # Tokenize
18
  inputs = tokenizer(texts, padding=True, truncation=True, return_tensors="tf")