Update app.py

app.py

@@ -10,23 +10,23 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    messages = f"""Generate a vivid story in 100-120 words using the following sentences: "{text}".
-    Structure requirements:
-    1. Only 1 paragraphs and the story mest have beginning, middle, end
-    2. Include at least two named characters
-    3. Use descriptive language
+    #messages = f"""Generate a vivid story in 100-120 words using the following sentences: "{text}".
+    #Structure requirements:
+    #1. Only 1 paragraphs and the story mest have beginning, middle, end
+    #2. Include at least two named characters
+    #3. Use descriptive language
     
-    Story:"""
+    #Story:"""
     #messages = f"""Generate a story of about 100 words using the following sentences: {text}."""
-    #messages = [
-    #    {"role": "system", "content": "You are a helpful AI assistant and will write stories based user input.The story must contain:Beginning, middle, end，Character development and around 100 words"},
-    #    {"role": "user", "content": text},
-    #]
-    text_to_story_model= pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
+    messages = [
+        {"role": "system", "content": "You are Qwen, created by Alibaba Cloud. You are a helpful assistant."},
+        {"role": "user", "content": f"""Please help generate a vivid story in 100-300 words using the following sentences: "{text}".The story must contain:Beginning, middle, end and use descriptive language"""},
+    ]
+    text_to_story_model= pipeline("text-generation", model="Qwen/Qwen2.5-0.5B-Instruct")
     story_text = text_to_story_model(
         messages,
-        max_length=150,  
-        min_length=50,
+        max_length=400,  
+        min_length=100,
         do_sample=True,
         temperature=0.65
     )[0]['generated_text']