Update app.py

app.py

@@ -10,11 +10,13 @@ def generate_image_caption(image):
     result = img2caption(image)
     return result[0]['generated_text']
 
-def text2story(text):
-    """Generates a story based on the input text using a pre-trained model."""
-    pipe = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2")
-    story_text = pipe(text)[0]['generated_text']
-    return story_text
+def text2story(text): 
+ text_to_story_model = pipeline("text-generation", model="pranavpsv/genre-story-generator-v2") 
+ story_text = text_to_story_model(text, max_new_tokens=150)[0]['generated_text'] 
+   words = story_text.split() 
+ if len(words) > 100: 
+ story_text = ' '.join(words[:100]) + '.' 
+   return story_text 
 
 def text2speech(text):
     """Converts text to speech using a pre-trained model."""