mkoot007 commited on
Commit
c8aeee2
·
1 Parent(s): d673a22

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ if generate_button:
17
  if title and character and punchline:
18
  prompt = f"Write a {genre} story titled '{title}' set in the {era} era. The main character, {character}, faces a {genre} adventure. The story is about '{punchline}'."
19
 
20
- set_seed(42) # Set a seed for reproducibility
21
  full_story = text_generator(prompt, max_length=story_length, do_sample=True)[0]["generated_text"]
22
 
23
  st.markdown(full_story)
 
17
  if title and character and punchline:
18
  prompt = f"Write a {genre} story titled '{title}' set in the {era} era. The main character, {character}, faces a {genre} adventure. The story is about '{punchline}'."
19
 
20
+ set_seed(42)
21
  full_story = text_generator(prompt, max_length=story_length, do_sample=True)[0]["generated_text"]
22
 
23
  st.markdown(full_story)