Update app.py

app.py

@@ -783,6 +783,16 @@ class Upscale:
                     print(f"> Processing image {gallery_idx:02d}, Shape: {img_cv2.shape}")
                     bg_upsample_img = None
                     if upscale_model and self.realesrganer and hasattr(self.realesrganer, "enhance"):
+                        # ---------------- SAFETY GUARD ----------------
+                        h, w = img_cv2.shape[:2]
+                        MAX_OUTPUT_PIXELS = 80_000_000
+                        out_pixels = h * w * (self.scale ** 2)
+                    
+                        if out_pixels > MAX_OUTPUT_PIXELS:
+                            raise RuntimeError(
+                                f"Image too large for upscale: {h}x{w} @ x{self.scale}"
+                            )
+                        # ----------------------------------------------
                         bg_upsample_img, _ = auto_split_upscale(img_cv2, self.realesrganer.enhance, self.scale) if is_auto_split_upscale else self.realesrganer.enhance(img_cv2, outscale=self.scale)
                         current_progress += progressRatio/progressTotal
                         if progress: