Update app.py
Browse files
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, ):
|