Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -133,10 +133,9 @@ all_input = all_input.rstrip('\\n')
133
 
134
 
135
  with st.expander("配置"):
136
- stop_tokens = ["<eot>", "\n\nHuman"]
137
  def cut_message(answer: str):
138
  end = []
139
- for etk in stop_tokens:
140
  offset = answer.find(etk)
141
  if offset > 0:
142
  end.append(offset)
 
133
 
134
 
135
  with st.expander("配置"):
 
136
  def cut_message(answer: str):
137
  end = []
138
+ for etk in ["<eot>", "\n\nHuman"]:
139
  offset = answer.find(etk)
140
  if offset > 0:
141
  end.append(offset)