Update app.py

app.py

@@ -9,7 +9,7 @@ def img2text(url):
 def text2story(text):
     story_generator = pipeline("text-generation", model="openai-community/gpt2")
     prompt = f"Create a short story under 100 words based on: {text}"
-    generated = story_generator(prompt, max_length=300, truncation=True)
+    generated = story_generator(prompt, max_length=100)
     story_text = generated[0]['generated_text']
     return story_text
 
@@ -40,8 +40,4 @@ if uploaded_file is not None:
     audio_data =text2audio(story)
 
     if st.button("Story Time!"):
-        st.audio(audio_data['audio'],
-                    format="audio/wav",
-                    start_time=0,
-                    sample_rate = audio_data['sampling_rate'])
-        st.audio(audio_data)
+        st.audio(audio_data['audio'], format="audio/wav", start_time=0, sample_rate = audio_data['sampling_rate'])