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 |
|