Update app.py

app.py

@@ -2,7 +2,7 @@ import tensorflow as tf
 from keras.utils import custom_object_scope
 import gradio as gr
 import numpy as np
-from PIL import Image, ImageDraw  # Importe o módulo ImageDraw
+from PIL import Image, ImageDraw
 
 # Defina a camada personalizada FixedDropout
 class FixedDropout(tf.keras.layers.Dropout):
@@ -55,13 +55,14 @@ def classify_image(inp):
     # Converte a imagem de saída de volta para numpy array
     output_image = np.array(output_image)
 
-    return predicted_class, output_image
+    # Retorna uma lista com a imagem de entrada e o rótulo
+    return [inp, output_image], predicted_class
 
 # Crie uma interface Gradio
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.inputs.Image(shape=(192, 256)),
-    outputs=[gr.outputs.Label(), gr.outputs.Image(type="numpy")],
+    outputs=[gr.outputs.Multi([gr.outputs.Image(type="numpy"), gr.outputs.Image(type="numpy")]), gr.outputs.Label()],
     capture_session=True,
     title="Detecção de Catarata",
     description="Esta interface permite a detecção de catarata em imagens de olhos.",