Upload app.py

app.py

@@ -63,7 +63,11 @@ def process_image(input_image, resolution_choice):
         input_image = resize_image(input_image, 768)
         processing_resolution = None
     else:
-        processing_resolution = 0  # 使用原始分辨率
+        if input_image.size[0] > 2560 or input_image.size[1] > 2560:
+            processing_resolution = 2560  # 限制最大分辨率
+            input_image = resize_image(input_image, 2560)
+        else:
+            processing_resolution = 0  # 使用原始分辨率
 
     # 处理图像
     pipe_out = pipe(
@@ -102,7 +106,7 @@ def create_gradio_interface():
                     value="768",  # 默认选择原始分辨率
                 )
                 gr.Markdown(
-                    "Select the resolution for processing the image. Higher resolution may take longer to process. 768 is recommended for faster processing and stable performance."
+                    "Select the resolution for processing the image, 768 is recommended for faster processing and stable performance. Higher resolution may take longer to process, we restrict the maximum resolution to 2560."
                 )
                 submit_btn = gr.Button("Remove Reflection", variant="primary")
             with gr.Column():