DHEIVER commited on
Commit
6a61f29
1 Parent(s): b1ed862

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -53,13 +53,16 @@ def predict(input_image):
53
  # Determine a classe mais provável
54
  predicted_class = class_names[np.argmax(prediction)]
55
 
56
- return predicted_class
 
 
 
57
 
58
  # Crie uma interface Gradio para fazer previsões
59
  iface = gr.Interface(
60
  predict,
61
  inputs=gr.inputs.Image(label="Carregue uma imagem da região ocular", type="pil"),
62
- outputs=gr.outputs.Label(num_top_classes=2, label="Avaliação"),
63
  theme="auto", # Usar o tema padrão do Gradio
64
  )
65
 
 
53
  # Determine a classe mais provável
54
  predicted_class = class_names[np.argmax(prediction)]
55
 
56
+ # Adicione informações adicionais à saída
57
+ confidence = prediction[0][np.argmax(prediction)] # Confiança na previsão
58
+
59
+ return f"Classificação: {predicted_class}\nConfiança: {confidence:.2f}"
60
 
61
  # Crie uma interface Gradio para fazer previsões
62
  iface = gr.Interface(
63
  predict,
64
  inputs=gr.inputs.Image(label="Carregue uma imagem da região ocular", type="pil"),
65
+ outputs=gr.outputs.Textbox(label="Resultado"),
66
  theme="auto", # Usar o tema padrão do Gradio
67
  )
68