Hugging Face's logo

Spaces:
chaseharmon
/
Rap-Verse-Generation-Demo
Sleeping

App Files Community
chaseharmon commited on
Commit
ac75dba
·
1 Parent(s): fc4a409

fixed some things

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -57,5 +57,5 @@ if question:
57
  model_inputs = inputs.to('cuda')
58
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
59
  response = tokenizer.batch_decode(generated_ids)
60
- display_placeholder.write(response)
61
 
 
57
  model_inputs = inputs.to('cuda')
58
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
59
  response = tokenizer.batch_decode(generated_ids)
60
+ display_placeholder.write(response[0])
61