Update app.py

app.py

@@ -34,7 +34,7 @@ def process_image(image, prompt):
     plt.imsave("mask.png", torch.sigmoid(preds))
     mask_image = Image.open("mask.png").convert("RGB")
 
-    # Convert image to BGR format
+    image = cv2.cvtColor(np.array(image))
     imageio.imwrite("./data/data.png", image)
     # Convert mask to grayscale format
     mask_image = mask_image.resize((image.shape[1], image.shape[0]))