Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
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.
|
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 |
|