weijiawu commited on
Commit
c8cfbb8
1 Parent(s): f82f439

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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,(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]
@@ -305,5 +305,5 @@ with gr.Blocks(
305
  outputs=[chatbot, state, click_state, image_input, mask_save_path],
306
  show_progress=False, queue=True)
307
 
308
- iface.queue(concurrency_count=1, api_open=False, max_size=10)
309
  iface.launch(server_name="0.0.0.0", enable_queue=True)
 
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]
 
305
  outputs=[chatbot, state, click_state, image_input, mask_save_path],
306
  show_progress=False, queue=True)
307
 
308
+ iface.queue(concurrency_count=3, api_open=False, max_size=10)
309
  iface.launch(server_name="0.0.0.0", enable_queue=True)