Spaces:

rayadav1005
/

text_generator

Sleeping

Ankit Yadav commited on May 11

Commit

8002dec

•

1 Parent(s): 7fb9c4d

Jarvis Model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,7 +35,8 @@ async def generate1(prompt):
         formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     output = ""
     for response in stream:
-        output += f'{response.token.text}+'
     yield output

         formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=True)
     output = ""
     for response in stream:
+        if response.token.text != '</s>':
+            output += response.token.text
     yield output