Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -57,7 +57,7 @@ def generate_text(prompt):
|
|
57 |
top_k=40,
|
58 |
repetition_penalty=1.1
|
59 |
)
|
60 |
-
results =
|
61 |
return results
|
62 |
|
63 |
def merlin_chat(message, history):
|
|
|
57 |
top_k=40,
|
58 |
repetition_penalty=1.1
|
59 |
)
|
60 |
+
results = tokenizer.decode(outputs[0], clean_up_tokenization_spaces=False) #.replace(prompt,"")
|
61 |
return results
|
62 |
|
63 |
def merlin_chat(message, history):
|