Update app.py

app.py

@@ -164,9 +164,9 @@ def imsize(img, max_size=512, maintain_aspect_ratio=True):
 
 
 def rem_glass(input_img):   
-    # w_0, h_0 = input_img.size
+    w_0, h_0 = input_img.size
     #resizing 
-    im = imsize(input_img, max_size=512, maintain_aspect_ratio=True)
+    im = imsize(input_img, max_size=256, maintain_aspect_ratio=False)
 
     width, height = im.size
     ori_im = im.copy()
@@ -190,7 +190,7 @@ def rem_glass(input_img):
     # img = upscale(img, model_cran_v2)
 
     # scale image to original size
-    img = img.resize((width, height))
+    img = img.resize((w_0, h_0))
     img.save("removal.png")
 
     inference_app(