Orcawise commited on
Commit
086e901
1 Parent(s): 6669a8d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def generate_text(prompt):
45
  )
46
  print("show the output", outputs)
47
  # Decode the generated tokens
48
- generated_text = tokenizer.decode(outputs, skip_special_tokens=True)[0]
49
  print("show the generated text", generated_text)
50
  return generated_text
51
  #############
 
45
  )
46
  print("show the output", outputs)
47
  # Decode the generated tokens
48
+ generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
49
  print("show the generated text", generated_text)
50
  return generated_text
51
  #############