Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def run_tinystyler_batch(source_texts, target_texts_batch, reranking, temperatur
|
|
58 |
max_length=1024
|
59 |
)
|
60 |
|
61 |
-
generated_texts = tokenizer.
|
62 |
return generated_texts
|
63 |
|
64 |
def run_tinystyler(source_text, target_texts, reranking, temperature, top_p):
|
|
|
58 |
max_length=1024
|
59 |
)
|
60 |
|
61 |
+
generated_texts = tokenizer.batch_decode(output, skip_special_tokens=True)
|
62 |
return generated_texts
|
63 |
|
64 |
def run_tinystyler(source_text, target_texts, reranking, temperature, top_p):
|