Update app.py

app.py

@@ -1,5 +1,3 @@
-pip install -r requirements.txt
-
 # import part
 import streamlit as st
 from transformers import pipeline
@@ -16,8 +14,11 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_pipeline = pipeline("text-generation", model="microsoft/Phi-4-mini-instruct", trust_remote_code=True)
-    story_text = story_pipeline(text, max_length=200)[0]['generated_text']
+    # 创建文本生成的 pipeline 对象
+    story_pipeline = pipeline("text-generation", model="qihoo360/TinyR1-32B-Preview")
+    # 调用 pipeline 生成故事文本，设置最大长度为 200
+    result = story_pipeline(text, max_length=200)
+    story_text = result[0]['generated_text']
     return story_text
 
 # text2audio
@@ -30,7 +31,7 @@ def text2audio(story_text):
 
 # main part
 st.set_page_config(page_title="Your Image to Audio Story",
-                   page_icon="🦜") # prepare configuration
+                   page_icon="🦜")  # prepare configuration
 st.header("Turn Your Image to Audio Story")
 
 # Upload image
@@ -46,21 +47,20 @@ if uploaded_file is not None:
     st.image(uploaded_file, caption="Uploaded Image",
              use_column_width=True)
 
-    #Stage 1: Image to Text
+    # Stage 1: Image to Text
     st.text('Processing img2text...')
     scenario = img2text(uploaded_file.name)
     st.write(scenario)
 
-    #Stage 2: Text to Story
+    # Stage 2: Text to Story
     st.text('Generating a story...')
     story = text2story(scenario)
     st.write(story)
 
-    #Stage 3: Story to Audio data
+    # Stage 3: Story to Audio data
     st.text('Generating audio data...')
     audio_data = text2audio(story)
 
     # Play button
     if st.button("Play Audio"):
-        st.audio(audio_data,
-                 format="audio/mpeg")
+        st.audio(audio_data, format="audio/mpeg")