Update app.py

app.py

@@ -80,7 +80,7 @@ def get_prompt(chat_input, click_state):
     }
     return prompt
 
-def chat_with_points(chat_input, click_state, state, mask_save_path,image_input):
+def chat_with_points(chat_input, click_state, state, mask,image_input):
     
     points, labels, captions = click_state
     
@@ -93,7 +93,7 @@ def chat_with_points(chat_input, click_state, state, mask_save_path,image_input)
 
 
     pipe = pipe
-    mask = cv2.imread(mask_save_path)
+    # mask = cv2.imread(mask_save_path)
     
     image_input = np.array(image_input)
     h,w = image_input.shape[:2]
@@ -142,7 +142,7 @@ def inference_seg_cap(image_input, point_prompt, language, sentiment, factuality
     text = "edit"
     image_input = create_bubble_frame(image_input, text, (evt.index[0], evt.index[1]))
 
-    yield state, state, click_state, image_input, out["mask_save_path"]
+    yield state, state, click_state, image_input, input_mask
 #     if not args.disable_gpt and model.text_refiner:
 #         refined_caption = model.text_refiner.inference(query=text, controls=controls, context=out['context_captions'])
 #         # new_cap = 'Original: ' + text + '. Refined: ' + refined_caption['caption']