LogicGoInfotechSpaces commited on
Commit
0c727e4
·
verified ·
1 Parent(s): 2530a89

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -783,6 +783,8 @@ class Upscale:
783
  print(f"> Processing image {gallery_idx:02d}, Shape: {img_cv2.shape}")
784
  bg_upsample_img = None
785
  if upscale_model and self.realesrganer and hasattr(self.realesrganer, "enhance"):
 
 
786
  # ---------------- SAFETY GUARD ----------------
787
  h, w = img_cv2.shape[:2]
788
  MAX_OUTPUT_PIXELS = 80_000_000
 
783
  print(f"> Processing image {gallery_idx:02d}, Shape: {img_cv2.shape}")
784
  bg_upsample_img = None
785
  if upscale_model and self.realesrganer and hasattr(self.realesrganer, "enhance"):
786
+ # --- DOWN-SCALE SAFETY FIX ---
787
+ img_cv2 = downscale_image_max_1000(img_cv2, max_size=1000)
788
  # ---------------- SAFETY GUARD ----------------
789
  h, w = img_cv2.shape[:2]
790
  MAX_OUTPUT_PIXELS = 80_000_000