Add max_res parameter.

app.py

@@ -40,11 +40,12 @@ def get_model(model_name, scale):
 
 
 def inference(img, scale_str, model_name):
+ max_res = 1024
  scale = int(scale_str.replace('x', ''))
  width, height = img.size
  print(width, height)
- if width > 2048 or height > 2048:
- img = img.thumbnail((2048, 2048), Image.ANTIALIAS)
+ if width > max_res or height > max_res:
+ img = img.thumbnail((max_res, max_res), Image.ANTIALIAS)
  model = get_model(model_name, scale)
  try:
  inputs = ImageLoader.load_image(img)