Update app.py

app.py

@@ -48,7 +48,7 @@ def classify_image(image):
     # Restituisci le percentuali di confidenza per ciascuna classe
     confidence_scores = {class_labels[i]: float(prediction[i]) for i in range(len(class_labels))}
     
-    return class_labels[predicted_class_idx]
+    return confidence_scores
 
 # Interfaccia di Gradio
 gr.Interface(