Update app.py

app.py

@@ -6,14 +6,18 @@ import os
 # function part
 # img2text
 def img2text(image_path):
-    image_to_text = pipeline("image-to-text", model="noamrot/FuseCap_Image_Captioning")
+    image_to_text = pipeline("image-to-text", model="Maciel/Muge-Image-Caption")
     text = image_to_text(image_path)[0]["generated_text"]
     return text
 
 # text2story
 def text2story(text):
+    messages = [
+    {"role": "user", "content": "Who are you?"},
+    ]
     # Using a smaller text generation model
-    generator = pipeline('text-generation', model='gpt2')
+    generator = pipeline("text-generation", model="mlx-community/Llama-3.2-1B-Instruct-4bit")
+    generator(messages)
     
     # Create a prompt for the story generation
     prompt = f"Write a fun children's story based on this: {text}. Once upon a time, "
@@ -34,8 +38,10 @@ def text2story(text):
     story_text = story_text.replace(prompt, "Once upon a time, ")
     
     # Make sure the story is at least 100 words
-    if len(story_text.split()) < 100:
-        story_text += " The children had a wonderful adventure and learned that imagination can take you anywhere. They returned home with smiles, eager to share their magical story with family and friends. And they lived happily ever after."
+    words = story_text.split()
+    if len(words) > 100:
+        # Simply truncate to 100 words
+        story_text = " ".join(words[:100])
     
     return story_text