Colby commited on
Commit
d3217a9
·
verified ·
1 Parent(s): 9496358

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ def merlin_chat(message, history):
144
  for attempt in range(3):
145
  # result = model(prompt, max_new_tokens=250, return_full_text=False, handle_long_generation="hole")
146
  result = generate_text(prompt, model_path, parameters, headers)
147
- response = result[0]['generated_text']
148
  start = 0
149
  end = 0
150
  cleanStr = response.lstrip()
 
144
  for attempt in range(3):
145
  # result = model(prompt, max_new_tokens=250, return_full_text=False, handle_long_generation="hole")
146
  result = generate_text(prompt, model_path, parameters, headers)
147
+ response = result[0]
148
  start = 0
149
  end = 0
150
  cleanStr = response.lstrip()