Update tasks/text.py

tasks/text.py

@@ -2,12 +2,15 @@ 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
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
 
+#additional imports 
+from transformers import pipeline
+import logging
+
 router = APIRouter()
 
 DESCRIPTION = "Random Baseline"
@@ -62,6 +65,8 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     available_pipeline = pipeline(tasks="text_classfication")
     print(available_pipeline)
+    logging.log(INFO, available_pipeline)
+    
     predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
 
     #--------------------------------------------------------------------------------------------