Omnibus commited on
Commit
ddd1ec8
·
verified ·
1 Parent(s): 0b4bdb1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -72,7 +72,8 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
72
  for response in stream:
73
  output += response.token.text
74
  if history:
75
- yield "",[[history],[(prompt,output)]]
 
76
  else:
77
  yield "",[(prompt,output)]
78
 
@@ -91,7 +92,7 @@ def generate(prompt, history,max_new_tokens,temperature=temperature,top_p=top_p,
91
  print(line)
92
 
93
  if history:
94
- return "",[[history],[(prompt,output)]]
95
  else:
96
  return "",[(prompt,output)]
97
 
 
72
  for response in stream:
73
  output += response.token.text
74
  if history:
75
+ print (history)
76
+ yield "",[history,(prompt,output)]
77
  else:
78
  yield "",[(prompt,output)]
79
 
 
92
  print(line)
93
 
94
  if history:
95
+ return "",[history,(prompt,output)]
96
  else:
97
  return "",[(prompt,output)]
98