nodes/modelscope-scepter
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #58 from modelscope/v1.2.0_dev

Browse Source 
V1.2.0 dev
This commit is contained in:
jiangzeyinzi
2024-11-01 21:30:20 +08:00
committed by GitHub
parent d9b207cf5b fe3e11b49e
commit edb46a615c
1 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scepter/studio/chatbot/chatbot.py
+11 -11
View File
@@ -74,7 +74,6 @@ class ChatBotUI(object):
cfg_file=self.model_choices[self.model_name])
self.pipe = ACEInference()
self.pipe.init_from_cfg(model_cfg)
self.retry_msg = ''
self.max_msgs = 20
self.enable_i2v = cfg.get('ENABLE_I2V', False)
@@ -176,6 +175,7 @@ class ChatBotUI(object):
self.history = gr.State(value=[])
self.images = gr.State(value={})
self.history_result = gr.State(value={})
self.retry_msg = gr.State(value='')
with gr.Group():
with gr.Row(equal_height=True):
with gr.Column(visible=True) as self.chat_page:
@@ -595,7 +595,7 @@ class ChatBotUI(object):
video_fps,
video_seed,
progress=gr.Progress(track_tqdm=True)):
self.retry_msg = message
retry_msg = message
gen_id = get_md5(message)[:12]
save_path = os.path.join(self.cache_dir, f'{gen_id}.png')
@@ -768,7 +768,8 @@ class ChatBotUI(object):
history.pop(0)
return history, images, history_result, self.get_history(
history), gr.update(value=''), gr.update(visible=False)
history), gr.update(value=''), gr.update(
visible=False), retry_msg
chat_inputs = [
self.extend_prompt, self.history, self.images, self.use_history,
@@ -781,7 +782,7 @@ class ChatBotUI(object):
chat_outputs = [
self.history, self.images, self.history_result, self.chatbot,
self.text, self.gallery
self.text, self.gallery, self.retry_msg
]
self.chat_btn.click(run_chat,
@@ -792,12 +793,11 @@ class ChatBotUI(object):
inputs=[self.text] + chat_inputs,
outputs=chat_outputs)
########################################
def retry_chat(*args):
return run_chat(self.retry_msg, *args)
def retry_fn(*args):
return run_chat(*args)
self.retry_btn.click(retry_chat,
inputs=chat_inputs,
self.retry_btn.click(retry_fn,
inputs=[self.retry_msg] + chat_inputs,
outputs=chat_outputs)
########################################
@@ -852,7 +852,7 @@ class ChatBotUI(object):
history = [(prompt,
f'{pre_info} The generated image is:\n {img_str}')]
return self.get_history(history), gr.update(value=''), gr.update(
visible=False)
visible=False), gr.update(value=-1)
with self.eg:
self.example_task = gr.Text(label='Task Name',
@@ -878,7 +878,7 @@ class ChatBotUI(object):
self.example_task, self.example_image, self.example_mask,
self.example_ref_im1, self.text, self.seed
],
outputs=[self.chatbot, self.text, self.gallery],
outputs=[self.chatbot, self.text, self.gallery, self.seed],
examples_per_page=4,
run_on_click=True)