Omnibus commited on
Commit
c1ae534
·
verified ·
1 Parent(s): 010002c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -240,7 +240,7 @@ def generate(prompt, history, agent_name=agents[0], sys_prompt="", temperature=0
240
  yield '', [(prompt,output)],summary[0],json_obj, json_hist,html_out
241
 
242
  if not title:
243
- for line in output:
244
  if "title" in line.lower() and ":" in line.lower():
245
  title = line.split(":")[1]
246
  print(f'title:: {title}')
 
240
  yield '', [(prompt,output)],summary[0],json_obj, json_hist,html_out
241
 
242
  if not title:
243
+ for line in output.split("\n"):
244
  if "title" in line.lower() and ":" in line.lower():
245
  title = line.split(":")[1]
246
  print(f'title:: {title}')