ms1449 commited on
Commit
b827b44
·
verified ·
1 Parent(s): 6ff10e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ if st.button("Generate Blog Post"):
22
  inputs_encoded = tokenizer.encode(prompt, return_tensors="pt")
23
 
24
  # Generate text
25
- model_output = model.generate(inputs_encoded, max_new_tokens=200, do_sample=True, temperature=0.7)
26
 
27
  # Decode the output
28
  output = tokenizer.decode(model_output, skip_special_tokens=True)
 
22
  inputs_encoded = tokenizer.encode(prompt, return_tensors="pt")
23
 
24
  # Generate text
25
+ model_output = model.generate(inputs_encoded, max_new_tokens=100, do_sample=True, temperature=0.7)
26
 
27
  # Decode the output
28
  output = tokenizer.decode(model_output, skip_special_tokens=True)