Update app.py

app.py

@@ -10,9 +10,14 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_text = ""   # to be completed
+    #story_text = ""   # to be completed
+    messages = [
+    {"role": "user", "content": "根据以下关键词生成一个100字的故事："+text },
+    ]
+    text_to_story_model= pipeline("text-generation", model="deepseek-ai/DeepSeek-R1", trust_remote_code=True)
+    story_text = text_to_story_model(messages)
     return story_text
-
+    
 # text2audio
 def text2audio(story_text):
     audio_data = ""     # to be completed
@@ -38,8 +43,8 @@ if uploaded_file is not None:
 
     #Stage 2: Text to Story
     st.text('Generating a story...')
-    #story = text2story(scenario)
-    #st.write(story)
+    story = text2story(scenario)
+    st.write(story)
 
     #Stage 3: Story to Audio data
     #st.text('Generating audio data...')