Spaces:
Runtime error
Runtime error
Update app.py
#1
by
divi7007 - opened
app.py
CHANGED
|
@@ -39,7 +39,7 @@ def generate(prompt, seq_len=128, temperature=0.8, filter_thres=0.9):
|
|
| 39 |
use_tqdm=True,
|
| 40 |
)
|
| 41 |
|
| 42 |
-
decoded_output = tokenizer.batch_decode(output_tensor, skip_special_tokens=True)
|
| 43 |
|
| 44 |
return decoded_output
|
| 45 |
|
|
|
|
| 39 |
use_tqdm=True,
|
| 40 |
)
|
| 41 |
|
| 42 |
+
decoded_output = tokenizer.batch_decode(output_tensor[0], skip_special_tokens=True)
|
| 43 |
|
| 44 |
return decoded_output
|
| 45 |
|