Spaces:
Sleeping
Sleeping
Update app.py
Browse filesright now the output is coming in square brackets so
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 |
|