ready1234

tasks/text.py

@@ -31,6 +31,9 @@ if not os.path.exists("tasks/model"):
     #download model for text task
     download(repo_id = "kantundpeterpan/frugal-ai-toy", dst = "tasks/model")
 
+print("### Model Dir")
+print(os.listdir("./tasks/model"))
+
 import random
 
 from .utils.evaluation import TextEvaluationRequest
@@ -94,7 +97,7 @@ async def evaluate_text(request: TextEvaluationRequest):
     print(os.listdir("./tasks"))
     
     #load model
-    model = model = load("model/tfidf_rf.skops", trusted = unknown)
+    model = model = load("tasks/model/tfidf_rf.skops", trusted = unknown)
     
     # Make predictions
     true_labels = test_dataset["label"]