Spaces:
Sleeping
Sleeping
Commit
·
9ca39e4
1
Parent(s):
afb665f
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ def generate_response(text, k = 70, p = 0.9, nb = 4):
|
|
23 |
|
24 |
generated_text = tokenizerM.decode(gneration[0], skip_special_tokens=True)
|
25 |
bayts = generated_text.split("[BSEP]")
|
26 |
-
while("
|
27 |
bayts = bayts[:-1]
|
28 |
bayts = bayts[:-1]
|
29 |
temp_poem = ''
|
|
|
23 |
|
24 |
generated_text = tokenizerM.decode(gneration[0], skip_special_tokens=True)
|
25 |
bayts = generated_text.split("[BSEP]")
|
26 |
+
while("FSEP" not in bayts[-1]):
|
27 |
bayts = bayts[:-1]
|
28 |
bayts = bayts[:-1]
|
29 |
temp_poem = ''
|