Update app.py

app.py

@@ -18,7 +18,7 @@ except ImportError:
 caption_model = pipeline("image-to-text", model="unography/blip-large-long-cap")
 
 # Load the GPT-2 model for story generation
-story_generator = pipeline("text-generation", model="openai-community/gpt2")
+story_generator = pipeline("text-generation", model="gpt2")
 
 def generate_caption(image):
     # Generate the caption for the uploaded image
@@ -27,7 +27,7 @@ def generate_caption(image):
 
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
-    prompt = f"Startinh with 'Once upon a time', based on the image described as '{caption}', here is a short and interesting story for children aged 3-10. The story is positive and happy in tone, with added imagination:\n\n"
+    prompt = f"Starting with 'Once upon a time', based on the image described as '{caption}', here is a short and interesting story for children aged 3-10. The story is positive and happy in tone, with added imagination:\n\n"
     story = story_generator(prompt, max_length=500, num_return_sequences=1)[0]["generated_text"]
     
     # Extract the story text from the generated output