Update app.py

app.py

@@ -127,15 +127,7 @@ def inference_seg_cap(image_input, point_prompt, language, sentiment, factuality
 
     out = model.inference(image_input, prompt, controls)
     state = state + [(None, "Image point: {}, Input label: {}".format(prompt["input_point"], prompt["input_label"]))]
-    # for k, v in out['generated_captions'].items():
-    #     state = state + [(f'{k}: {v}', None)]
-#     state = state + [("caption: {}".format(out['generated_captions']['raw_caption']), None)]
-#     wiki = out['generated_captions'].get('wiki', "")
-#     click_state[2].append(out['generated_captions']['raw_caption'])
-    
-#     text = out['generated_captions']['raw_caption']
-    # draw = ImageDraw.Draw(image_input)
-    # draw.text((evt.index[0], evt.index[1]), text, textcolor=(0,0,255), text_size=120)
+
     input_mask = np.array(out['mask'].convert('P'))
     image_input = mask_painter(np.array(image_input), input_mask)
     origin_image_input = image_input
@@ -143,13 +135,6 @@ def inference_seg_cap(image_input, point_prompt, language, sentiment, factuality
     image_input = create_bubble_frame(image_input, text, (evt.index[0], evt.index[1]))
 
     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']
-#         new_cap = refined_caption['caption']
-#         refined_image_input = create_bubble_frame(origin_image_input, new_cap, (evt.index[0], evt.index[1]))
-#         yield state, state, click_state, chat_input, refined_image_input, wiki
-
 
 def upload_callback(image_input, state):
     state = [] + [('Image size: ' + str(image_input.size), None)]