Update app.py

app.py

@@ -159,15 +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]) > 1000 or max(img.shape[:2])>1500: # avoid memory exceeded due to too large img resolution
+        if min(img.shape[:2]) > 1100 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
+            face_upsample = False
+
+        h, w = img.shape[:2]
+        if h * w > 4_000_000: # avoid memory exceeded due to too large img resolution
             raise gr.Error(
-                "Image resolution is too large (short side > 1200 or long side > 2000). "
+                "Image resolution is too large (>4 megapixels). "
                 "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."
             )