Update app.py

app.py

@@ -48,7 +48,7 @@ def text2story(text):
         story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
         
         # Generate the story with the superhero genre
-        prompt = f"<BOS> <superhero> {text}"  # Add genre tags to the prompt
+        prompt = f"<BOS> <fun superhero> {text}"  # Add genre tags to the prompt
         story = story_generator(prompt, max_length=150, num_return_sequences=1)[0]['generated_text']
         
         # Remove <BOS> and <superhero> tags from the generated story