Update app.py

app.py

@@ -15,8 +15,17 @@ with open("pca_model.pkl", "rb") as f:
 with open("kmeans_model.pkl", "rb") as f:
     model_cluster = pickle.load(f)
 
-with open("lg.pkl", "rb") as f:
-    model_diabetes = pickle.load(f)
+
+
+# Vérifie si le fichier existe et charge-le
+try:
+    with open("lg.pkl", "rb") as f:
+        model_diabetes = pickle.load(f)
+        print("Model loaded successfully!")
+except FileNotFoundError:
+    print("File 'lg.pkl' not found.")
+except EOFError:
+    print("File 'lg.pkl' is empty or corrupted.")
 
 # Charger le dataframe (assurez-vous de mettre le bon chemin vers votre fichier)
 df_scaled = pd.read_csv("data.csv")  # Remplacez par le chemin réel du fichier
@@ -80,11 +89,11 @@ with gr.Blocks() as app:
 
     # **Section Clustering**
     gr.Markdown("### Visualisation des Clusters des Réactions en Ligne")
-    
+
     # Calculate unique clusters after PCA and prediction
     unique_clusters = pca_df['Cluster'].unique()
-    
-    cluster_selector = gr.Dropdown(["Tous"] + [str(i) for i in unique_clusters], 
+
+    cluster_selector = gr.Dropdown(["Tous"] + [str(i) for i in unique_clusters],
                                    label="Sélectionner un cluster")
     cluster_plot = gr.Plot()