KaiChen1998 commited on
Commit
b849db2
·
1 Parent(s): 0d4bcf1

fix regenerate

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,8 +88,8 @@ def load_demo_refresh_model_list():
88
 
89
  def regenerate(state, image_process_mode):
90
  logging.info(f"regenerate.")
91
- state.messages[-1][-1] = None
92
- prev_human_msg = state.messages[-2]
93
  if type(prev_human_msg[1]) in (tuple, list):
94
  prev_human_msg[1] = (*prev_human_msg[1][:2], image_process_mode, *prev_human_msg[1][3:])
95
  state.skip_next = False
 
88
 
89
  def regenerate(state, image_process_mode):
90
  logging.info(f"regenerate.")
91
+ state.messages = state.messages[:-3]
92
+ prev_human_msg = state.messages[-1]
93
  if type(prev_human_msg[1]) in (tuple, list):
94
  prev_human_msg[1] = (*prev_human_msg[1][:2], image_process_mode, *prev_human_msg[1][3:])
95
  state.skip_next = False