ArmelR commited on
Commit
74d6bba
1 Parent(s): 7703eca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -155,7 +155,7 @@ def generate(
155
  if len(past_messages) < 1:
156
  prompt = get_inference_prompt(messages=[{"role": "user", "content": user_message}], model_name=model_name)
157
  else:
158
- prompt = dialogue_template.get_inference_prompt(messages=past_messages + [{"role": "user", "content": user_message}], model_name=model_name)
159
 
160
  generate_kwargs = {
161
  "temperature": temperature,
 
155
  if len(past_messages) < 1:
156
  prompt = get_inference_prompt(messages=[{"role": "user", "content": user_message}], model_name=model_name)
157
  else:
158
+ prompt = get_inference_prompt(messages=past_messages + [{"role": "user", "content": user_message}], model_name=model_name)
159
 
160
  generate_kwargs = {
161
  "temperature": temperature,