Update evaluate.py
Browse files- 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("
|
15 |
-
model = TFAutoModelForSequenceClassification.from_pretrained("
|
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")
|