Update app.py

app.py

@@ -17,8 +17,6 @@ except ImportError:
 # Load the image captioning model
 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="gpt2")
 story_generator = pipeline("text-generation", model="distilbert/distilgpt2")
 
 def generate_caption(image):
@@ -26,24 +24,9 @@ def generate_caption(image):
     caption = caption_model(image)[0]["generated_text"]
     return caption
 
-#def generate_story(caption):
-    # Generate the story based on the caption using the GPT-2 model
-    #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
-    #story = story.split("\n\n")[1].strip()
-    
-    #return story
-
-#def generate_story(caption):
-    # Generate the story based on the caption
-    #story = story_generator(caption, max_length=200, num_return_sequences=1)[0]["generated_text"]
-    #return story
-
 def generate_story(caption):
     # Generate the story based on the caption using the GPT-2 model
-    prompt = f"Once upon a time, based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their imagination:\n\n"
+    prompt = f"Startig with 'Once upon a time...', also, based on the image described as '{caption}', here is a short, simple, and engaging story for children aged 3-10. The story should be easy to understand, use age-appropriate language, and convey a positive message. Focus on the main elements in the image and create a story that sparks their 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
@@ -61,7 +44,6 @@ def generate_story(caption):
     
     return story
     
-
 def convert_to_audio(story):
     # Convert the story to audio using gTTS
     tts = gTTS(text=story, lang="en")