Update app.py

app.py

@@ -44,11 +44,12 @@ def classify_image_with_overlay(img):
 iface = gr.Interface(
     fn=classify_image_with_overlay,
     inputs=gr.inputs.Image(),
-    outputs=gr.outputs.Image(type="numpy"),
+    outputs=gr.outputs.Image(type="numpy", output_width=400, output_height=400),  # Ajuste o tamanho da saída desejado
     live=True,
     title="ViT Image Classifier with Overlay",
     description="Upload an image for classification with label overlay.",
 )
 
+
 if __name__ == "__main__":
     iface.launch()