Update app.py

app.py

@@ -30,14 +30,20 @@ def classify_image(image):
     # Rótulo da classe prevista
     predicted_label = class_labels[predicted_class]
     
-    # Retorna o rótulo da classe prevista
-    return predicted_label
+    # Formate a saída em HTML
+    result_html = f"""
+    <h2>Resultado da Classificação</h2>
+    <p><strong>Classe Predita:</strong> {predicted_label}</p>
+    """
+    
+    # Retorna o resultado formatado em HTML
+    return result_html
 
 # Crie uma interface Gradio
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.inputs.Image(),
-    outputs="text",  # Apenas o rótulo da classe prevista
+    outputs=gr.outputs.HTML(),  # Saída formatada com HTML
     title="Classificador de Tumor da Tireoide",
     description="Faça o upload de uma imagem de um tumor da tireoide para classificação.",
 )