Update app.py

app.py

@@ -40,7 +40,9 @@ def predict_image(input_image):
     prediction = loaded_model.predict(input_image)
 
     # A saída será uma matriz de previsões (no caso de classificação de duas classes, será algo como [[probabilidade_classe_0, probabilidade_classe_1]])
-    return prediction
+    # Adicione lógica para interpretar o resultado com base no texto gerado
+    text = "Normal" if prediction[0][0] > prediction[0][1] else "Cataract"
+    return text
 
 # Crie uma interface Gradio para fazer previsões
 iface = gr.Interface(