downscale input images

https://huggingface.co/spaces/finegrain/finegrain-image-enhancer/discussions/7

src/app.py

@@ -118,8 +118,18 @@ def process(
     generator = torch.Generator(device=DEVICE)
     generator.manual_seed(seed)
 
+    # Resize to avoid using too much VRAM.
+    # If you have a bug GPU you can go higher.
+    side_size = min(input_image.size)
+    if side_size > 768:
+        scale = 768 / side_size
+        new_size = (int(input_image.width * scale), int(input_image.height * scale))
+        resized_image = input_image.resize(new_size, resample=Image.Resampling.LANCZOS)
+    else:
+        resized_image = input_image
+
     enhanced_image = enhancer.upscale(
-        image=input_image,
+        image=resized_image,
         prompt=prompt,
         negative_prompt=negative_prompt,
         upscale_factor=upscale_factor,