Update app.py

app.py

@@ -11,7 +11,7 @@ from PIL import Image
 vl_model = Qwen2VLForConditionalGeneration.from_pretrained(
     "Qwen/Qwen2-VL-2B-Instruct", torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32, device_map="auto"
 )
-vl_processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", max_pixels=480*480)
+vl_processor = AutoProcessor.from_pretrained("Qwen/Qwen2-VL-2B-Instruct", max_pixels=240*240)
 
 # Load Text Model
 model_name = "Qwen/Qwen2.5-Math-1.5B-Instruct"
@@ -96,7 +96,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             with gr.Tabs() as input_tabs:
                 with gr.Tab("Upload"):
-                    input_image = gr.Image(type="pil", label="Upload")
+                    input_image = resize_image(gr.Image(type="pil", label="Upload"))
                 with gr.Tab("Sketch"):
                     input_sketchpad = gr.Sketchpad(type="pil", label="Sketch", layers=False)
             input_tabs.select(fn=tabs_select, inputs=[state])