Update app.py

app.py

@@ -55,7 +55,19 @@ def predict_image(input_image):
     predicted_class = class_names[np.argmax(prediction)]
     probability = prediction[0][np.argmax(prediction)]
 
-    formatted_text = f"Predicted Class: {predicted_class}\nProbability: {probability:.2%}"
+    # 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>
+"""
     return formatted_text
 
 # Crie uma interface Gradio para fazer previsões
@@ -63,7 +75,7 @@ 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="default"
+    interpretation="html"
 )
 
 # Execute a interface Gradio