Spaces:
Sleeping
Sleeping
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")
|