Update app.py

app.py

@@ -13,9 +13,9 @@ def img2text(image):
 # Function to generate a story based on the caption
 def text2story(text):
     # Load the text generation model
-    story_model = pipeline("text-generation", model="distilbert/distilgpt2")
+    story_model = pipeline("text-generation", model="gpt2")
     # Generate a story based on the input text
-    story_text = story_model(f"Once upon a time, {text}.", max_length=100, num_return_sequences=1)
+    story_text = story_model(f"Once upon a time, {text}.", max_length=100, num_return_sequences=1, do_sample=True, top_k=50)
     return story_text[0]["generated_text"]  # Return the generated story
 
 # Function to convert text to audio