Update app.py

app.py

@@ -14,7 +14,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_pipeline = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
+    story_pipeline = pipeline("text-generation", model="GSAI-ML/LLaDA-8B-Instruct", trust_remote_code=True)
     story_text = story_pipeline(text, max_length=200)[0]['generated_text']
     return story_text
 
@@ -61,5 +61,4 @@ if uploaded_file is not None:
     # Play button
     if st.button("Play Audio"):
         st.audio(audio_data,
-                    format="audio/wav",
-                    start_time=0)
+                 format="audio/mpeg")