Colby commited on
Commit
ee57cda
·
verified ·
1 Parent(s): e99b177

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -142,7 +142,7 @@ def merlin_chat(message, history):
142
  prompt_data = history
143
  prompt_data.append(user_msg)
144
  prompt_data.insert(0,system_msg)
145
- prompt = json.dumps(system_msg)[:-1] + ",{'role': 'assistant', 'content': '"
146
  for attempt in range(3):
147
  # result = model(prompt, max_new_tokens=250, return_full_text=False, handle_long_generation="hole")
148
  result = generate_text(prompt, model_path, parameters, headers)
 
142
  prompt_data = history
143
  prompt_data.append(user_msg)
144
  prompt_data.insert(0,system_msg)
145
+ prompt = json.dumps(system_msg)[:-1] + ",{\"role\": \"assistant\", \"content\": \""
146
  for attempt in range(3):
147
  # result = model(prompt, max_new_tokens=250, return_full_text=False, handle_long_generation="hole")
148
  result = generate_text(prompt, model_path, parameters, headers)