updated code

models/model_tfidf_linearsvc.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1b362701d616def1dffcd59059ce4c9e9ef8f53561ffebe5225eafaf65b70362
+size 326963

tasks/text.py

@@ -2,7 +2,7 @@ from fastapi import APIRouter
 from datetime import datetime
 from datasets import load_dataset
 from sklearn.metrics import accuracy_score
-import random
+import pickle
 
 from .utils.evaluation import TextEvaluationRequest
 from .utils.emissions import tracker, clean_emissions_data, get_space_info
@@ -55,10 +55,14 @@ async def evaluate_text(request: TextEvaluationRequest):
     # YOUR MODEL INFERENCE CODE HERE
     # 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)
+
+    # Charger le modèle sauvegardé avec pickle
+    with open('../models/model_grid_search.pkl', 'rb') as f:
+        model_loaded = pickle.load(f)
+
+    # Make predictions (placeholder for actual model inference)
     true_labels = test_dataset["label"]
-    predictions = [random.randint(0, 7) for _ in range(len(true_labels))]
+    predictions = model_loaded.predict(test_dataset["quote"])
 
     #--------------------------------------------------------------------------------------------
     # YOUR MODEL INFERENCE STOPS HERE