Update app.py

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
 from transformers import pipeline, set_seed
 
-# Create a text2text-generation pipeline with the "google/flan-t5-base" model
 text_generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
 st.title("Story Generator")
@@ -16,10 +15,8 @@ generate_button = st.button("Generate Story")
 
 if generate_button:
     if title and character and punchline:
-        # Prepare a structured prompt for the story
         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}'."
 
-        # Generate the story
         set_seed(42)  # Set a seed for reproducibility
         full_story = text_generator(prompt, max_length=story_length, do_sample=True)[0]["generated_text"]