Update app.py

app.py

@@ -54,9 +54,6 @@ def text2story(text):
         # Remove <BOS> and <superhero> tags from the generated story
         story = story.replace("<BOS>", "").replace("<superhero>", "").strip()
         
-        # Remove the input text (scenario) from the generated story to avoid redundancy
-        if text in story:
-            story = story.replace(text, "").strip()
         
         # Ensure the story is within 100 words by truncating if necessary
         story = " ".join(story.split()[:100])
@@ -65,8 +62,6 @@ def text2story(text):
         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("<superhero>", "").strip()
-            if text in story:
-                story = story.replace(text, "").strip()
             story = " ".join(story.split()[:100])
         
         return story