Update app.py

app.py

@@ -63,10 +63,12 @@ def calc_psnr(pred, target):
 def standard_upscale(img):
     if img is None: return None, ""
         
-    max_input_dim = 1024 
+    # Bumped max input to 2K (2048px) for a 4K output
+    max_input_dim = 2048 
     w, h = img.size
     
     if w > max_input_dim or h > max_input_dim:
+        gr.Warning(f"Input image exceeded the 2K ({max_input_dim}px) limit. It has been proportionally downscaled to ensure the 4K output fits in server memory.")
         scale = max_input_dim / max(w, h)
         w, h = int(w * scale), int(h * scale)
         img = img.resize((w, h), Image.BICUBIC)
@@ -101,11 +103,12 @@ def benchmark_upscale(hr_img):
     h = h - (h % 2)
     hr_img = hr_img.crop((0, 0, w, h))
     
-    max_input_dim = 2048 # HR can be 2048 because LR will be 1024
-    if w > max_input_dim or h > max_input_dim:
-        scale = max_input_dim / max(w, h)
+    # HR can now be 4K (4096px) because the LR input to the model will be max 2K
+    max_hr_dim = 4096 
+    if w > max_hr_dim or h > max_hr_dim:
+        gr.Warning(f"Ground truth image exceeded the 4K ({max_hr_dim}px) limit. It has been proportionally downscaled.")
+        scale = max_hr_dim / max(w, h)
         w, h = int(w * scale), int(h * scale)
-        # Ensure even dimensions again after resize
         w = w - (w % 2)
         h = h - (h % 2)
         hr_img = hr_img.resize((w, h), Image.BICUBIC)
@@ -142,14 +145,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
     gr.Markdown(
         """
         # ⚡ FastEDSR 2x Image Upscaler
-        Upload an image to enhance and upscale it by 2x. 
+        Upload an image to enhance and upscale it by 2x. Supports up to 4K resolution output.
         """
     )
     
     with gr.Tabs():
         # TAB 1: STANDARD
         with gr.TabItem("⚡ Standard Upscaling"):
-            gr.Markdown("Directly upscale any low-resolution image.")
+            gr.Markdown("Directly upscale any low-resolution image. Inputs over 2K (2048px) will be scaled down to prevent memory limits.")
             with gr.Row():
                 with gr.Column():
                     std_input = gr.Image(type="pil", label="Low Resolution Input")
@@ -162,7 +165,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as app:
 
         # TAB 2: BENCHMARK
         with gr.TabItem("📊 Benchmark Mode"):
-            gr.Markdown("Upload a high-quality image. The app will compress it, upscale it, and measure the PSNR quality against the original.")
+            gr.Markdown("Upload a high-quality image (up to 4K). The app will compress it to 2x lower resolution, upscale it using FastEDSR, and measure the PSNR quality against the original.")
             with gr.Row():
                 with gr.Column():
                     bm_input = gr.Image(type="pil", label="Ground Truth (High Res) Image")