Update app.py

app.py

@@ -54,18 +54,18 @@ def predict_image(input_image):
     class_names = ["Normal", "Cataract"]
     probabilities = prediction[0]
 
-    # Formate os resultados com todos os rótulos e probabilidades
-    result_text = ""
+    # Use o Gradio para criar um gráfico de barras horizontais
+    bars = []
     for class_name, probability in zip(class_names, probabilities):
-        result_text += f"{class_name}: {probability:.2%}\n"
+        bars.append(gr.predict.Bar(label=class_name, confidence=probability))
 
-    return result_text
+    return bars
 
 # Crie uma interface Gradio para fazer previsões
 iface = gr.Interface(
     fn=predict_image,
     inputs="image",
-    outputs="text",
+    outputs=gr.outputs.Image(label="Predictions"),
     interpretation="default"
 )