Update app.py

app.py

@@ -32,7 +32,7 @@ def predict(image):
     output_image = (output_image * 255).astype(np.uint8)
     filtered_image = cv2.medianBlur(output_image, 3)
     output_masked = (processed_image * 255).astype(np.uint8)
-    return Image.fromarray(filtered_image), Image.fromarray(output_masked)
+    return Image.fromarray(output_masked), Image.fromarray(filtered_image)
 
 demo = gr.Interface(
     fn=predict,