Update app.py

app.py

@@ -55,28 +55,18 @@ def predict_image(input_image):
     predicted_class = class_names[np.argmax(prediction)]
     probability = prediction[0][np.argmax(prediction)]
 
-    # Altere a interpretação dos resultados para uma tabela HTML
-    formatted_text = f"""
-<table>
-  <tr>
-    <th>Predicted Class</th>
-    <th>Probability</th>
-  </tr>
-  <tr>
-    <td>{predicted_class}</td>
-    <td>{probability:.2%}</td>
-  </tr>
-</table>
-"""
+    # Formate os resultados da previsão para exibição usando o componente BarPlot
+    bar_plot_values = [probability, 1 - probability]
+    bar_plot_labels = ["Normal", "Cataract"]
+
+    formatted_text = gr.Plot(values=bar_plot_values, labels=bar_plot_labels, type="bar").render()
     return formatted_text
 
 # Crie uma interface Gradio para fazer previsões
 iface = gr.Interface(
     fn=predict_image,
     inputs=gr.inputs.Image(label="Upload an Image", type="pil"),
-    outputs=gr.outputs.Textbox(label="Prediction", type="text"),
-    interpretation="html",
-    custom_interpretation_fn=lambda prediction: formatted_text
+    outputs=gr.outputs.Textbox(label="Prediction", type="text")
 )
 
 # Execute a interface Gradio