fixed errors

tasks/text.py

@@ -67,8 +67,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     sentence_model = SentenceTransformer(model_name)
 
     # Convert each sentence into a vector representation (embedding)
-    embeddings = model.encode(test_dataset['quote'].tolist())
-    embeddings = embeddings.cpu()
+    embeddings = sentence_model.encode(test_dataset['quote'])
 
     # Update the code below to replace the random baseline by your model inference within the inference pass where the energy consumption and emissions are tracked.
     #--------------------------------------------------------------------------------------------   
@@ -91,11 +90,11 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     # Make predictions
     with torch.no_grad():
-        outputs = model_nn(text_embeddings)
+        outputs = model_nn(embeddings)
         _, predicted = torch.max(outputs, 1)  # Get the class with the highest score
-    
+
     # Decode the predictions back to original labels using label_encoder
-    predicted_labels = label_encoder.inverse_transform(predicted.cpu().numpy())
+    predictions = predicted.cpu().numpy()
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE