Update app.py

app.py

@@ -54,19 +54,16 @@ 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, probability
+    return predicted_class
 
 # 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), gr.outputs.ProgressBar()],
+    outputs=gr.outputs.Label(num_top_classes=2),
     title="Normal? Or Cataract?",
 )
 
 # Execute a interface Gradio
-iface.launch()
+iface.launch()