Update tasks/text.py

tasks/text.py

@@ -72,12 +72,12 @@ async def evaluate_text(request: TextEvaluationRequest):
     current_file_path = os.path.abspath(__file__)
     current_dir = os.path.dirname(current_file_path)
 
-    with open(os.path.join(current_dir,"tfidf_vectorizer2.pkl"), "rb") as tfidf_file:
-        tfidf_vectorizer = cloudpickle.load(tfidf_file)
+    """ with open(os.path.join(current_dir,"tfidf_vectorizer2.pkl"), "rb") as tfidf_file:
+        tfidf_vectorizer = cloudpickle.load(tfidf_file)"""
     
 
     # Make predictions using the loaded model
-    predictions = predict(test_dataset,tfidf_vectorizer,os.path.join(current_dir,"random_forest_model.pkl"))
+    predictions = predict(test_dataset,os.path.join(current_dir,"tfidf_vectorizer_params.json"),os.path.join(current_dir,"random_forest_model.pkl"))
     predictions = [LABEL_MAPPING[label] for label in predictions]
 
     #--------------------------------------------------------------------------------------------