Aliou12 commited on
Commit
f586f9f
·
verified ·
1 Parent(s): 8218a11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -10,13 +10,12 @@ import sys
10
  import pickle
11
 
12
  # Charger les modèles
13
- try:
14
- model = joblib.load("lg.pkl")
15
- model_cluster = joblib.load("kmeans_model.pkl")
16
- print("Modèles chargés avec succès.")
17
- except FileNotFoundError:
18
- print("Erreur: Le fichier n'a pas été trouvé.")
19
- exit(1)
20
 
21
  # Fonction de prédiction du diabète
22
  def predict_diabetes(pregnancies, glucose, blood_pressure, skin_thickness, insulin, bmi, dpf, age):
 
10
  import pickle
11
 
12
  # Charger les modèles
13
+
14
+ with open("lg.pkl", "rb") as f:
15
+ model = pickle.load(f)
16
+ with open("kmeans_model.pkl", "rb") as f:
17
+ model_cluster = pickle.load(f)
18
+
 
19
 
20
  # Fonction de prédiction du diabète
21
  def predict_diabetes(pregnancies, glucose, blood_pressure, skin_thickness, insulin, bmi, dpf, age):