Update app.py

app.py

@@ -99,7 +99,7 @@ def chat_with_points(chat_input, click_state, state, mask,image_input):
     h,w = image_input.shape[:2]
     
     image = cv2.resize(image_input,(512,512))
-    mask = cv2.resize(mask,(512,512)).astype(np.uint8)[:,:,0]
+    mask = cv2.resize(mask,(512,512)).astype(np.uint8)
     print(image.shape,mask.shape)
     print("chat_input:",chat_input)
     image = pipe(prompt=chat_input, image=image, mask_image=mask).images[0]