Update app.py

app.py

@@ -12,27 +12,12 @@ def img2text(image_path):
 
 # text2story
 def text2story(text):
-    messages = [
-    {"role": "user", "content": "Who are you?"},
-    ]
     # Using a smaller text generation model
-    generator = pipeline("text-generation", model="mlx-community/Llama-3.2-1B-Instruct-4bit")
-    generator(messages)
+    generator = pipeline("text-generation", model="TinyLlama/TinyLlama-1.1B-Chat-v1.0")
     
     # Create a prompt for the story generation
     prompt = f"Write a fun children's story based on this: {text}. Once upon a time, "
-    
-    # Generate the story
-    story_result = generator(
-        prompt,
-        max_length=200,
-        num_return_sequences=1,
-        temperature=0.8,
-        top_k=50,
-        top_p=0.95,
-        do_sample=True
-    )
-    
+   
     # Extract the generated text
     story_text = story_result[0]['generated_text']
     story_text = story_text.replace(prompt, "Once upon a time, ")