Update app.py

app.py

@@ -494,10 +494,6 @@ current_image = None
 current_predictions = None
 
 # ---- Fonctions pour l'UI Gradio ----
-import pandas as pd
-import numpy as np
-import gradio as gr
-
 def quick_predict_ui(image_pil):
     global current_image, current_predictions
     if image_pil is None:
@@ -567,7 +563,7 @@ def quick_predict_ui(image_pil):
             warning_visible = True
         
         # CORRECTION : Créer un DataFrame pandas pour le graphique
-        probabilities = [ensemble_probs[i] * 100 for i in range(len(CLASS_NAMES))]
+        probabilities = [round(ensemble_probs[i] * 100, 2) for i in range(len(CLASS_NAMES))]
         df_plot = pd.DataFrame({
             'Classes': CLASS_NAMES,
             'Probabilités (%)': probabilities
@@ -593,7 +589,7 @@ def quick_predict_ui(image_pil):
             None,  # Réinitialiser le graphique en cas d'erreur
             f"❌ Erreur: {str(e)}"
         )
-
+        
 def generate_gradcam_ui(progress=gr.Progress()):
     global current_image, current_predictions
     if current_image is None or current_predictions is None:
@@ -708,7 +704,8 @@ with gr.Blocks(theme=theme, title="Analyse de lésions", css=css) as demo:
                 y="Probabilités (%)",  # Nom de la colonne y
                 title="Probabilités par classe",
                 y_lim=[0, 100],
-                height=450
+                height=450,
+                show_label=True
             )
             
             output_gradcam = gr.Image(label="🔍 Visualisation Grad-CAM")