xzl12306 commited on
Commit
dc6e6a5
·
1 Parent(s): 8646f88
Files changed (1) hide show
  1. app.py +9 -6
app.py CHANGED
@@ -60,12 +60,15 @@ For an optimal experience, please use desktop computers for this demo, as mobile
60
  """
61
 
62
  def regenerate(state, image_process_mode):
63
- state.messages[-1][-1] = None
64
- prev_human_msg = state.messages[-2]
65
- if type(prev_human_msg[1]) in (tuple, list):
66
- prev_human_msg[1] = (*prev_human_msg[1][:2], image_process_mode)
67
- state.skip_next = False
68
- return (state, state.to_gradio_chatbot(), "", None)
 
 
 
69
 
70
 
71
  def clear_history():
 
60
  """
61
 
62
  def regenerate(state, image_process_mode):
63
+ if state is not None:
64
+ state.messages[-1][-1] = None
65
+ prev_human_msg = state.messages[-2]
66
+ if type(prev_human_msg[1]) in (tuple, list):
67
+ prev_human_msg[1] = (*prev_human_msg[1][:2], image_process_mode)
68
+ state.skip_next = False
69
+ return (state, state.to_gradio_chatbot(), "", None)
70
+ else:
71
+ return (None, "", "", None)
72
 
73
 
74
  def clear_history():