Spaces:
Sleeping
Sleeping
Ankit Yadav
commited on
Commit
·
8002dec
1
Parent(s):
7fb9c4d
Jarvis Model
Browse files
app.py
CHANGED
|
@@ -35,7 +35,8 @@ async def generate1(prompt):
|
|
| 35 |
formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
|
| 36 |
output = ""
|
| 37 |
for response in stream:
|
| 38 |
-
|
|
|
|
| 39 |
|
| 40 |
yield output
|
| 41 |
|
|
|
|
| 35 |
formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
|
| 36 |
output = ""
|
| 37 |
for response in stream:
|
| 38 |
+
if response.token.text != '</s>':
|
| 39 |
+
output += response.token.text
|
| 40 |
|
| 41 |
yield output
|
| 42 |
|