Update app.py

app.py

@@ -30,7 +30,7 @@ def predict(image):
     input_image = np.expand_dims(processed_image, axis=0)
     output_image = model.predict(input_image)[0]
     output_image = (output_image * 255).astype(np.uint8)
-    filtered_image = cv2.medianBlur(output_image, kernel_size=3)
+    filtered_image = cv2.medianBlur(output_image, 3)
     output_masked = (processed_image * 255).astype(np.uint8)
     return Image.fromarray(filtered_image), Image.fromarray(output_masked)