Update app.py

app.py

@@ -12,17 +12,13 @@ def img2text(img):
 
 # text2story
 def text2story(text):
-    prompt = "Once upon a time in a land far, far away"
-
     text_generation_model = pipeline("text-generation",
-                     model="distilbert/distilgpt2")
-
-    generated_story = text_generation_model(prompt,
+                                   model="meta-llama/Meta-Llama-3-8B")
+    story_text = text_generation_model(text)[0]["generated_text"]
+    generated_story = text_generation_model(story_text,
                             max_length=100,
                             num_return_sequences=1)
-
-print("The story is here......")
-print(generated_story[0]['generated_text'])
+    return generated_story
         
 # text2audio
 def text2audio(story_text):