Update tasks/text.py

tasks/text.py

@@ -64,7 +64,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     train_test = dataset["train"].train_test_split(test_size=request.test_size, seed=request.test_seed)
     test_dataset = train_test["test"]
     # Batch embed all test dataset quotes
-    test_embeddings = embedding_model.encode(test_dataset["quote"], batch_size=128, convert_to_numpy=True, normalize_embeddings=True)
+    test_embeddings = embedding_model.encode(test_dataset["quote"], batch_size=128)
 
     # Start tracking emissions
     tracker.start()
@@ -81,7 +81,7 @@ async def evaluate_text(request: TextEvaluationRequest):
 
     # Apply threshold (0.9) for classification
     predictions = [
-        LABEL_MAPPING[class_labels[idx]] if sim > 0.9 else LABEL_MAPPING["0_not_relevant"]
+        LABEL_MAPPING[class_labels[idx]] if sim > 0.8 else LABEL_MAPPING["0_not_relevant"]
         for idx, sim in zip(best_indices, best_similarities)
     ]