Spaces:
Running
Running
Commit
•
b0d7588
1
Parent(s):
b064d79
add rsplit to avoid eos token on model output
Browse files
app.py
CHANGED
|
@@ -52,8 +52,9 @@ def generate_emoji_translation(prompt):
|
|
| 52 |
|
| 53 |
for response in stream:
|
| 54 |
output += response.token.text
|
|
|
|
| 55 |
|
| 56 |
-
return output
|
| 57 |
|
| 58 |
|
| 59 |
with gr.Blocks() as demo:
|
|
|
|
| 52 |
|
| 53 |
for response in stream:
|
| 54 |
output += response.token.text
|
| 55 |
+
yield output.rsplit("<", 1)[0]
|
| 56 |
|
| 57 |
+
return output.rsplit("<", 1)[0]
|
| 58 |
|
| 59 |
|
| 60 |
with gr.Blocks() as demo:
|