Update app.py

app.py

@@ -39,17 +39,19 @@ def classify_image_with_overlay(img):
     # Put the label text on the white rectangle
     cv2.putText(image, label, org, font, font_scale, (0, 0, 0), thickness, cv2.LINE_AA)
 
+    # Resize the image to a larger size
+    image = cv2.resize(image, (800, 800))  # Defina o tamanho desejado aqui
+
     return image
 
 iface = gr.Interface(
     fn=classify_image_with_overlay,
     inputs=gr.inputs.Image(),
-    outputs=gr.outputs.Image(type="numpy", output_width=400, output_height=400),  # Ajuste o tamanho da saída desejado
+    outputs=gr.outputs.Image(type="numpy"),
     live=True,
     title="ViT Image Classifier with Overlay",
     description="Upload an image for classification with label overlay.",
 )
 
-
 if __name__ == "__main__":
     iface.launch()