Downscale to prevent OOM

#2
by pcuenq HF staff - opened
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ processor = FuyuProcessor(image_processor=FuyuImageProcessor(), tokenizer=tokeni
15
 
16
  caption_prompt = "Generate a coco-style caption.\\n"
17
 
18
- def resize_to_max(image, max_width=1920, max_height=1080):
19
  width, height = image.size
20
  if width <= max_width and height <= max_height:
21
  return image
 
15
 
16
  caption_prompt = "Generate a coco-style caption.\\n"
17
 
18
+ def resize_to_max(image, max_width=1080, max_height=1080):
19
  width, height = image.size
20
  if width <= max_width and height <= max_height:
21
  return image