Update app.py

app.py

@@ -44,14 +44,14 @@ def classify_image(input_image):
     label_text = f"Predicted Class: {predicted_class}"
     draw.text((10, 10), label_text, (255, 0, 0), font=font)
 
-    return output_image, class_probabilities
+    return input_image[0], output_image, class_probabilities
 
 # Crie uma interface Gradio
 input_interface = gr.Interface(
     fn=classify_image,
     inputs="image",  # Especifique o tipo de entrada como "image"
-    outputs=["image", "text"],  # Especifique dois tipos de saída: imagem e texto
-    output_labels=["Output Image", "Class Probabilities"]  # Rotule as saídas
+    outputs=["image", "image", "text"],  # Especifique três tipos de saída: imagem de entrada, imagem de saída e texto
+    output_labels=["Input Image", "Output Image", "Class Probabilities"]  # Rotule as saídas
 )
 
 # Inicie o aplicativo Gradio