Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def generate(
|
|
| 62 |
)
|
| 63 |
|
| 64 |
generated_segment = output_ids[0][input_ids.shape[1]:]
|
| 65 |
-
return tokenizer.decode(generated_segment, skip_special_tokens=
|
| 66 |
|
| 67 |
|
| 68 |
demo = gr.ChatInterface(
|
|
|
|
| 62 |
)
|
| 63 |
|
| 64 |
generated_segment = output_ids[0][input_ids.shape[1]:]
|
| 65 |
+
return tokenizer.decode(generated_segment, skip_special_tokens=True)
|
| 66 |
|
| 67 |
|
| 68 |
demo = gr.ChatInterface(
|