Update app.py

app.py

@@ -41,7 +41,12 @@ A value of 0.1 means to repaint wherever the model is certain it's a watermark.
 
 def fix_img(img0, threshold=0.5, erosion=2, dilation=6, mask_only=False,
             meaningful_diff=150, meaningful_ed=2):
-    img = img0.convert('RGB')
+    sz_max = 2048
+    scale = min(sz_max / img0.size[0], sz_max / img0.size[1])
+    if scale < 0.99:
+        img = img0.convert('RGB').resize((int(img0.size[0] * scale), int(img0.size[1] * scale)))
+    else:
+        img = img0.convert('RGB')
     npa = np.array(img)
     blue = npa[:, :, 2]
     blue_torch = torch.tensor(blue).float().to(device)
@@ -70,7 +75,7 @@ def fix_img(img0, threshold=0.5, erosion=2, dilation=6, mask_only=False,
     fixed_r = np.where(mask, fixed_r, npa[:, :, 0])
     fixed_g = np.where(mask, fixed_g, npa[:, :, 1])
     fixed_b = np.where(mask, fixed_b, npa[:, :, 2])
-    
+
     if meaningful_diff > 0:
         diff_r = np.abs(fixed_r - npa[:, :, 0])
         diff_g = np.abs(fixed_g - npa[:, :, 1])
@@ -103,7 +108,7 @@ with gr.Blocks() as demo:
         erosion = gr.Slider(minimum=1, maximum=10, value=2, label="Erosion", step=1)
         dilation = gr.Slider(minimum=1, maximum=10, value=10, label="Dilation", step=1)
         mask_only = gr.Checkbox(value=False, label="Mask Only")
-        meaningful_diff = gr.Slider(minimum=0, maximum=255, value=50, label="Meaningful Diff", step=1)
+        meaningful_diff = gr.Slider(minimum=0, maximum=128, value=50, label="Meaningful Diff", step=1)
         meaningful_ed = gr.Slider(minimum=1, maximum=10, value=2, label="Meaningful ED", step=1)
 
     # Outputs Section