ms1449 commited on
Commit
0e090f2
·
verified ·
1 Parent(s): 4f5142e

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
  generation_config = GenerationConfig(max_new_tokens=50, do_sample=True, temperature=0.7)
23
 
24
  # Tokenize the input
25
- input_ids = tokenizer.encode(prompt, return_tensors="pt")
26
 
27
  # Model output
28
  model_output = model.generate(inputs_encoded["input_ids"], generation_config=generation_config)[0]
 
22
  generation_config = GenerationConfig(max_new_tokens=50, do_sample=True, temperature=0.7)
23
 
24
  # Tokenize the input
25
+ input_encoded = tokenizer.encode(prompt, return_tensors="pt")
26
 
27
  # Model output
28
  model_output = model.generate(inputs_encoded["input_ids"], generation_config=generation_config)[0]