Update app.py

app.py

@@ -11,12 +11,12 @@ def generate_image_caption(image):
     return result[0]['generated_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 
+    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."""