Update app.py

app.py

@@ -12,19 +12,20 @@ def classificar_imagem(file_name):
     pred = new_model.predict(np.array([img]))
     pred = np.round(pred, 1)
     if pred == 0:
-        pred = "Benigno"
+        pred_label = "Benigno"
     else:
-        pred = "Maligno"
-    return pred
+        pred_label = "Maligno"
+    pred_score = f"{float(pred) * 100:.2f}%"
+    return pred_label, pred_score
 
 
-imagem = gr.inputs.File(file_count=1, label="Arquivo a ser analisado (extensão .pgm)")
-texto_saida = gr.outputs.Textbox(label="Resultado do diagnóstico")
+imagem = gr.inputs.File(file_count=1, label="Imagem de mamografia")
+resultado_diagnostico = [gr.outputs.Textbox(label="Diagnóstico"), gr.outputs.Textbox(label="Confiança")]
 
 gr.Interface(
     fn=classificar_imagem,
     inputs=imagem,
-    outputs=texto_saida,
+    outputs=resultado_diagnostico,
     interpretation="default",
     live=True,
     theme="dark-peach",