chaseharmon commited on
Commit
e92315a
·
1 Parent(s): 9ca4cad
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -52,4 +52,5 @@ if question:
52
  model_inputs = inputs.to('cuda')
53
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
54
  decoded_output = tokenizer.batch_decode(generated_ids)
 
55
 
 
52
  model_inputs = inputs.to('cuda')
53
  generated_ids = model.generate(**model_inputs, max_new_tokens=300, do_sample=True, pad_token_id=tokenizer.eos_token_id)
54
  decoded_output = tokenizer.batch_decode(generated_ids)
55
+ st.write(decoded_output)
56