juste load pickle sans la fonction

BiGHeaDMaX · Nov 6, 2023 · db3d0dc · db3d0dc
1 parent 6b193f6
commit db3d0dc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/main.py b/app/main.py
@@ -12,8 +12,8 @@
 app = FastAPI()
 
 # Chargement du modèle choisi précédemment enregistré
-#with open('KNeighborsClassifier_and_bin.pkl', 'rb') as fichier:
-#    model_and_bin = pickle.load(fichier)
+with open('KNeighborsClassifier_and_bin.pkl', 'rb') as fichier:
+    model_and_bin = pickle.load(fichier)
 
 # Chargement du modèle USE pré-entraîné (s'il n'avait pas été chargé précédemment)
 # embed = hub.load("https://tfhub.dev/google/universal-sentence-encoder/4")