Update app.py

app.py

@@ -10,7 +10,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_generator = pipeline("text-generation", model="distilgpt2")  # Corrected pipeline initialization
+    story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator")  # Corrected pipeline initialization
     story_text = story_generator(text, max_length=150, num_return_sequences=1)  # Pass parameters here
     return story_text[0]["generated_text"]  # Extract generated text