modified text.py

tasks/text.py

@@ -6,6 +6,15 @@ import random
 
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
+from transformers import pipeline
+
+# load model from huggingface.co/models using our repository id
+classifier = pipeline(
+    task="text-classification",
+    model="axel-darmouni/test_modernbert_finetuned",
+    tokenizer="answerdotai/ModernBERT-base",
+)
+
 
 router = APIRouter()
 
@@ -58,7 +67,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     
     # Make random predictions (placeholder for actual model inference)
     true_labels = test_dataset["label"]
-    predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
+    predictions = [classifier(sample)["label"] for sample in test_dataset["quote"]]
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE