weijiawu commited on
Commit
0bfd6ca
1 Parent(s): c8cfbb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,8 +98,8 @@ def chat_with_points(chat_input, click_state, state, mask_save_path,image_input)
98
  image_input = np.array(image_input)
99
  h,w = image_input.shape[:2]
100
 
101
- image = cv2.resize(image_input,(256,256))
102
- mask = cv2.resize(mask,(256,256)).astype(np.uint8)[:,:,0]
103
  print(image.shape,mask.shape)
104
  print("chat_input:",chat_input)
105
  image = pipe(prompt=chat_input, image=image, mask_image=mask).images[0]
 
98
  image_input = np.array(image_input)
99
  h,w = image_input.shape[:2]
100
 
101
+ image = cv2.resize(image_input,(512,512))
102
+ mask = cv2.resize(mask,(512,512)).astype(np.uint8)[:,:,0]
103
  print(image.shape,mask.shape)
104
  print("chat_input:",chat_input)
105
  image = pipe(prompt=chat_input, image=image, mask_image=mask).images[0]