MatthiasPicard commited on
Commit
84167fd
·
verified ·
1 Parent(s): 4ea35be

Update tasks/text.py

Browse files
Files changed (1) hide show
  1. 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/modernBERT_0.75_acc_cards_rewritten'
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)