ECUiVADE commited on
Commit
faa2249
1 Parent(s): 155f1dd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -151,6 +151,7 @@ def generate(prompt, history):
151
  output = llm(context, max_tokens=400, stop=["Nurse:"], echo=False)
152
  response = output["choices"][0]["text"]
153
  response = response.strip()
 
154
 
155
 
156
  # for output in llm(input, stream=True, max_tokens=100, ):
 
151
  output = llm(context, max_tokens=400, stop=["Nurse:"], echo=False)
152
  response = output["choices"][0]["text"]
153
  response = response.strip()
154
+ yield response
155
 
156
 
157
  # for output in llm(input, stream=True, max_tokens=100, ):