Commit
·
e92315a
1
Parent(s):
9ca4cad
write
Browse files
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 |
|