Update app.py

app.py

@@ -11,7 +11,7 @@ def prever(imagem):
     predicao = str(predicao)
     probabilidades = [f"{labels[i]}: {prob * 100:.2f}%" for i, prob in enumerate(probabilidades)]
     
-    return predicao, probabilidades
+    return predicao  # Return only the predicted label
 
 titulo = "Detecção de Câncer de Mama com Aprendizado Profundo e Transferência de Aprendizado (ResNet18)"
 descricao = "<p style='text-align: center'><b>Como radiologista ou oncologista, é crucial saber o que há de errado com uma imagem de raio-X de mama.<b><br><b> Faça o upload da imagem de raio-X de mama para saber o que há de errado com o seio de uma paciente com ou sem implante<b><p>"
@@ -20,7 +20,7 @@ exemplos = [['img1.jpeg'], ['img2.jpeg']]
 habilitar_fila = True
 
 inputs = gr.inputs.Image(shape=(512, 512), label="Imagem de Raio-X de Mama")
-outputs = [gr.outputs.Label(label="Diagnóstico"), gr.outputs.Label(label="Probabilidades")]
+outputs = gr.outputs.Label(label="Diagnóstico")  # Update the output interface
 
 interface = gr.Interface(
     fn=prever,