Vision-CAIR commited on
Commit
cbab169
1 Parent(s): 0566ef0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -385,8 +385,9 @@ def gradio_reset(chat_state, img_list, path_list):
385
  chat_state.messages = []
386
  if img_list is not None:
387
  img_list = []
388
- for path in path_list:
389
- os.remove(path)
 
390
  return None, gr.update(value=None, interactive=True), gr.update(placeholder='Upload your image and chat',
391
  interactive=True), chat_state, img_list
392
 
@@ -486,7 +487,8 @@ def gradio_stream_answer(chatbot, chat_state, img_list, temperature):
486
  def gradio_visualize(chatbot, gr_img, path_list):
487
  if isinstance(gr_img, dict):
488
  gr_img, mask = gr_img['image'], gr_img['mask']
489
-
 
490
  unescaped = reverse_escape(chatbot[-1][1])
491
  visual_img, generation_color = visualize_all_bbox_together(gr_img, unescaped)
492
  if visual_img is not None:
 
385
  chat_state.messages = []
386
  if img_list is not None:
387
  img_list = []
388
+ if isinstance(path_list, list):
389
+ for path in path_list:
390
+ os.remove(path)
391
  return None, gr.update(value=None, interactive=True), gr.update(placeholder='Upload your image and chat',
392
  interactive=True), chat_state, img_list
393
 
 
487
  def gradio_visualize(chatbot, gr_img, path_list):
488
  if isinstance(gr_img, dict):
489
  gr_img, mask = gr_img['image'], gr_img['mask']
490
+ if not isinstance(path_list, list):
491
+ path_list = []
492
  unescaped = reverse_escape(chatbot[-1][1])
493
  visual_img, generation_color = visualize_all_bbox_together(gr_img, unescaped)
494
  if visual_img is not None: