Update tasks/text.py

tasks/text.py

@@ -12,9 +12,6 @@ import joblib
 REPO_ID = "kantundpeterpan/frugal-ai-toy"
 FILENAME = "tfidf_rf.pkl"
 
-model = joblib.load(
-    hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
-)
 
 import random
 
@@ -70,7 +67,12 @@ async def evaluate_text(request: TextEvaluationRequest):
     # 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.
     #--------------------------------------------------------------------------------------------   
 
-    # Make random predictions (placeholder for actual model inference)
+    #load model
+    model = joblib.load(
+        hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
+    )
+    
+    # Make predictions
     true_labels = test_dataset["label"]
     predictions = [
         LABEL_MAPPING[r] for r in model.predict(test_dataset)