AjayP13 commited on
Commit
1ab6216
·
verified ·
1 Parent(s): 58979a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.decode_batch(output, skip_special_tokens=True)
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):