Spaces:
Running
on
T4
Running
on
T4
Update app.py
Browse files
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
|