Update tasks/text.py
Browse files- tasks/text.py +1 -1
tasks/text.py
CHANGED
@@ -65,7 +65,7 @@ async def evaluate_text(request: TextEvaluationRequest):
|
|
65 |
# Make random predictions (placeholder for actual model inference)
|
66 |
true_labels = test_dataset["label"]
|
67 |
predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
|
68 |
-
path_model = 'MatthiasPicard/
|
69 |
path_tokenizer = "answerdotai/ModernBERT-base"
|
70 |
|
71 |
model = AutoModelForSequenceClassification.from_pretrained(path_model)
|
|
|
65 |
# Make random predictions (placeholder for actual model inference)
|
66 |
true_labels = test_dataset["label"]
|
67 |
predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
|
68 |
+
path_model = 'MatthiasPicard/checkpoint4200_batch16_modern_bert_valloss_0.79_0.74acc'
|
69 |
path_tokenizer = "answerdotai/ModernBERT-base"
|
70 |
|
71 |
model = AutoModelForSequenceClassification.from_pretrained(path_model)
|