Update app.py

app.py

@@ -145,11 +145,16 @@ 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 max(img.shape[:2]) > 1500: # avoid memory exceeded due to too large img resolution
-            upscale = 1
-            background_enhance = False
-            face_upsample = False
-
+        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
+            # face_upsample = False
+            raise gr.Error(
+                "Image resolution is too large (short side > 1100 or long side > 1500). "
+                "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."
+            )
+            
         face_helper = FaceRestoreHelper(
             upscale,
             face_size=512,
@@ -182,9 +187,10 @@ def inference(image, face_align, background_enhance, face_upsample, upscale, cod
         if min(img.shape[:2]) > 1000 and num_det_faces > 15:
             raise gr.Error(
                 "Too many faces detected (>15) in a high-resolution image. "
-                "To keep the demo responsive, this case is skipped. "
-                "For such inputs, please deploy the demo locally without this limit."
+                "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."
             )
+
             
         # face restoration for each cropped face
         for idx, cropped_face in enumerate(face_helper.cropped_faces):
@@ -326,7 +332,7 @@ demo = gr.Interface(
         gr.Slider(0, 1, value=0.5, step=0.01, label='Codeformer_Fidelity (0 for better quality, 1 for better identity)')
     ], [
         gr.Image(type="filepath", label="Output"),
-        gr.Markdown("Please download the output within 30 seconds.")
+        gr.Markdown("**Please download the output within 30 seconds.**")
     ],
     title=title,
     description=description,
@@ -339,7 +345,7 @@ demo = gr.Interface(
         ['05.jpg', True, True, True, 2, 0.1],
         ['06.png', False, True, True, 1, 0.5]
       ],
-    concurrency_limit=1
+    concurrency_limit=2
     )
 
 DEBUG = os.getenv('DEBUG') == '1'