Update app.py

app.py

@@ -79,7 +79,7 @@ def model_process(image, mask):
     original_shape = image.shape
     interpolation = cv2.INTER_CUBIC
     
-    size_limit = 1080 # "Original"
+    size_limit = 512 #1080 # "Original"
     if size_limit == "Original":
         size_limit = max(image.shape)
     else:
@@ -148,8 +148,8 @@ def predict(input):
         mask_pil = input['mask']
         image = np.array(image_pil)
         mask = np.array(mask_pil.convert("L"))
-        output = mask_pil 
-    # output = model_process(image, mask)
+        # output = mask_pil 
+    output = model_process(image, mask)
     return output
 
 css = '''