Niki Zhang commited on
Commit
acab436
·
verified ·
1 Parent(s): 1d4e620

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -19,7 +19,7 @@ from caption_anything.utils.parser import parse_augment
19
  from caption_anything.captioner import build_captioner
20
  from caption_anything.text_refiner import build_text_refiner
21
  from caption_anything.segmenter import build_segmenter
22
- from caption_anything.utils.chatbot import ConversationBot, build_chatbot_tools, get_new_image_name
23
  from segment_anything import sam_model_registry
24
  import easyocr
25
  import re
@@ -958,7 +958,7 @@ def create_ui():
958
 
959
  with gr.Blocks(
960
  css=get_style(),
961
- theme=gr.themes.Soft()
962
  ) as iface:
963
  state = gr.State([])
964
  out_state = gr.State(None)
@@ -1229,7 +1229,7 @@ def create_ui():
1229
 
1230
  mv_images = gr.State()
1231
 
1232
- chatbot.like(handle_like_dislike, inputs=[like_state, dislike_state], outputs=[like_state, dislike_state])
1233
 
1234
  submit.click(fn=check_input_image, inputs=[new_crop_save_path], outputs=[processed_image]).success(
1235
  fn=generate_mvs,
 
19
  from caption_anything.captioner import build_captioner
20
  from caption_anything.text_refiner import build_text_refiner
21
  from caption_anything.segmenter import build_segmenter
22
+ from chatbox import ConversationBot, build_chatbot_tools, get_new_image_name
23
  from segment_anything import sam_model_registry
24
  import easyocr
25
  import re
 
958
 
959
  with gr.Blocks(
960
  css=get_style(),
961
+ theme=gr.themes.Base()
962
  ) as iface:
963
  state = gr.State([])
964
  out_state = gr.State(None)
 
1229
 
1230
  mv_images = gr.State()
1231
 
1232
+ # chatbot.like(handle_like_dislike, inputs=[like_state, dislike_state], outputs=[like_state, dislike_state])
1233
 
1234
  submit.click(fn=check_input_image, inputs=[new_crop_save_path], outputs=[processed_image]).success(
1235
  fn=generate_mvs,