Update app.py

app.py

@@ -159,12 +159,15 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
             upscale = 4 
         if upscale > 2 and max(img.shape[:2])>1000: # avoid memory exceeded due to too large img resolution
             upscale = 2 
-        if min(img.shape[:2]) > 1100 or max(img.shape[:2]) > 1500: # avoid memory exceeded due to too large img resolution
+        if min(img.shape[:2]) > 1000 or max(img.shape[:2])>1500: # avoid memory exceeded due to too large img resolution
+            upscale = 1
+            background_enhance = False
+        if min(img.shape[:2]) > 1200 or max(img.shape[:2]) > 2000: # avoid memory exceeded due to too large img resolution
             # upscale = 1
             # background_enhance = False
             # face_upsample = False
             raise gr.Error(
-                "Image resolution is too large (short side > 1100 or long side > 1500). "
+                "Image resolution is too large (short side > 1200 or long side > 2000). "
                 "To keep the demo responsive and avoid long queue times, this case is skipped. "
                 "For such inputs, please deploy this demo locally and remove this limit."
             )