Update app.py

app.py

@@ -37,10 +37,10 @@ def text2story(text):
         str: Generated superhero story suitable for kids aged 3-10, within 100 words.
     """
     # Load the text generation model
-    story_generator = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
+    story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
     
     # Generate the story with the superhero genre
-    prompt = f"{text}"
+    prompt = f"<BOS> <superhero> {text}"
     story = story_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
     
     # Remove <BOS> and <superhero> tags from the generated story