Update tasks/text.py

tasks/text.py

@@ -76,7 +76,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     
 
     # Make predictions using the loaded model
-    predictions = predict(test_dataset,os.path.join(current_dir,"tfidf_vectorizer_params.json"),os.path.join(current_dir,"tfidf_vectorizer_vocab.pkl"),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,"tfidf_vectorizer_vocab.pkl"),os.path.join(current_dir,"tfidf_vectorizer_idf.pkl"),os.path.join(current_dir,"random_forest_model.pkl"))
     predictions = [LABEL_MAPPING[label] for label in predictions]
 
     #--------------------------------------------------------------------------------------------