Update app.py

app.py

@@ -30,16 +30,14 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2111.098
 
 def inference(img,task):
     os.system('mkdir temp')
-    max_res = 904
+    
+    # Resize the longer edge of the input image
+    max_res = 512
     width, height = img.size
     if max(width,height) > max_res:
-      scale = min(width,height)/max(width,height)
-      if width > max_res:
-        width = max_res
-        height = int(scale*max_res)
-      if height > max_res:
-        height = max_res
-        width = int(scale*max_res)
+      scale = max_res /max(width,height)
+      width = int(scale*width)
+      height = int(scale*height)
       img = img.resize((width,height), Image.ANTIALIAS)
   
     img.save("temp/image.jpg", "JPEG")