Upload app.py

app.py

@@ -45,12 +45,21 @@ def process_image(input_image):
     # 将 Gradio 输入转换为 PIL 图像
     input_image = Image.fromarray(input_image)
 
+    # 根据输入图像的大小设置分辨率
+    max_side = max(input_image.size)
+    if max_side < 768:
+        resolution = 768
+    elif max_side > 2560:
+        resolution = 2560
+    else:
+        resolution = 0
+
     # 处理图像
     pipe_out = pipe(
         image=input_image,
         prompt="remove glass reflection",
         vae_2=vae_2,
-        processing_resolution=None,
+        processing_resolution=resolution,
     )
 
     # 将输出转换为图像