Update tasks/text.py

tasks/text.py

@@ -2,6 +2,7 @@ from fastapi import APIRouter
 from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
+from transformers import pipeline
 import random
 
 from .utils.evaluation import TextEvaluationRequest
@@ -58,9 +59,10 @@ 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))]
 
-    print("meow")
+    available_pipeline = pipeline(tasks="text_classfication")
+    print(available_pipeline)
+    predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE