Update app.py

app.py

@@ -54,14 +54,17 @@ def predict(input_image):
 
     # Determine a classe mais provável
     predicted_class = class_names[np.argmax(prediction)]
+    
+    # Obtenha a probabilidade da classe prevista
+    probability = prediction[0][np.argmax(prediction)]
 
-    return predicted_class
+    return predicted_class, probability
 
 # Crie uma interface Gradio para fazer previsões
 iface = gr.Interface(
     predict,
     inputs=gr.inputs.Image(label="Upload hot dog candidate", type="pil"),
-    outputs=gr.outputs.Label(num_top_classes=2),
+    outputs=[gr.outputs.Label(num_top_classes=2), gr.outputs.ProgressBar()],
     title="Normal? Or Cataract?",
 )