Update app.py

app.py

@@ -48,11 +48,11 @@ def text2story(text):
         story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")
         
         # Generate the story with the superhero genre
-        prompt = f"<BOS> <funny superhero> {text}"  # Add genre tags to the prompt
+        prompt = f"<BOS> <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
-        story = story.replace("<BOS>", "").replace("<funny superhero>", "").strip()
+        story = story.replace("<BOS>", "").replace("<superhero>", "").strip()
         
         
         # Ensure the story is within 100 words by truncating if necessary
@@ -61,7 +61,7 @@ def text2story(text):
         # If the story is too short, regenerate it
         if len(story.split()) < 50:  # Minimum 50 words
             story = story_generator(prompt, max_length=200, num_return_sequences=1)[0]['generated_text']
-            story = story.replace("<BOS>", "").replace("<funny superhero>", "").strip()
+            story = story.replace("<BOS>", "").replace("<superhero>", "").strip()
             story = " ".join(story.split()[:100])
         
         return story