Limour commited on
Commit
6481b74
1 Parent(s): 2c53b67

Upload 3 files

Browse files
Files changed (3) hide show
  1. mods/btn_com.py +3 -1
  2. mods/btn_status_bar.py +4 -0
  3. mods/btn_vo.py +1 -1
mods/btn_com.py CHANGED
@@ -14,6 +14,7 @@ def init(cfg):
14
  cfg['btn_submit_vo_suggest'] = gr.Button("Submit&旁白&建议", variant="primary")
15
  cfg['btn_submit'] = gr.Button("Submit")
16
  cfg['btn_suggest'] = gr.Button("建议")
 
17
 
18
  cfg['btn_stop_status'] = True
19
 
@@ -90,7 +91,8 @@ def init(cfg):
90
 
91
  btn_start_or_finish_outputs = [cfg['btn_submit'], cfg['btn_vo'],
92
  cfg['btn_suggest'], cfg['btn_retry'],
93
- cfg['btn_submit_vo_suggest']]
 
94
 
95
  def btn_start_or_finish(finish):
96
  tmp = gr.update(interactive=finish)
 
14
  cfg['btn_submit_vo_suggest'] = gr.Button("Submit&旁白&建议", variant="primary")
15
  cfg['btn_submit'] = gr.Button("Submit")
16
  cfg['btn_suggest'] = gr.Button("建议")
17
+ cfg['btn_status_bar'] = gr.Button("状态")
18
 
19
  cfg['btn_stop_status'] = True
20
 
 
91
 
92
  btn_start_or_finish_outputs = [cfg['btn_submit'], cfg['btn_vo'],
93
  cfg['btn_suggest'], cfg['btn_retry'],
94
+ cfg['btn_submit_vo_suggest'],
95
+ cfg['btn_status_bar']]
96
 
97
  def btn_start_or_finish(finish):
98
  tmp = gr.update(interactive=finish)
mods/btn_status_bar.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ def init(cfg):
2
+ chat_template = cfg['chat_template']
3
+ model = cfg['model']
4
+ lock = cfg['session_lock']
mods/btn_vo.py CHANGED
@@ -30,7 +30,7 @@ def init(cfg):
30
  for _h in _tmp:
31
  yield _h, model.venv_info
32
  # ========== 及时清理上一次生成的旁白 ==========
33
- model.venv_remove('vo', keep_last=True)
34
  yield _h, model.venv_info
35
  print('清理旁白', model.venv_info)
36
 
 
30
  for _h in _tmp:
31
  yield _h, model.venv_info
32
  # ========== 及时清理上一次生成的旁白 ==========
33
+ model.venv_remove('vo', keep_last=1)
34
  yield _h, model.venv_info
35
  print('清理旁白', model.venv_info)
36