mkoot007 commited on
Commit
a7ef976
1 Parent(s): 406f121

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -19,8 +19,8 @@ if st.button("Generate Story"):
19
 
20
  # Generate the story
21
  set_seed(42) # Set a seed for reproducibility
22
- story = text_generator(prompt, max_length=300, do_sample=True)[0]["generated_text"]
23
 
24
- st.markdown(story)
25
  else:
26
  st.warning("Please provide a title and character name.")
 
19
 
20
  # Generate the story
21
  set_seed(42) # Set a seed for reproducibility
22
+ full_story = text_generator(prompt, max_length=1000, do_sample=True)[0]["generated_text"]
23
 
24
+ st.markdown(full_story)
25
  else:
26
  st.warning("Please provide a title and character name.")