Update app.py

app.py

@@ -7,9 +7,9 @@ def img2text(url):
     return text
 
 def text2story(text):
-    story_generator = pipeline("text-generation", model="pranavpsv/gpt2-genre-story-generator", device_map="auto", return_full_text=False)
+    story_generator = pipeline("text-generation", model="coffeeee/nsfw-story-generator2", device_map="auto", return_full_text=False)
     prompt = f"Write a story under 100 words based on: {text}."
-    generated = story_generator(prompt, max_new_tokens=150, do_sample=False, temperature=0.5)
+    generated = story_generator(prompt, max_new_tokens=140, do_sample=True, temperature=0.7)
     story_text = generated[0]['generated_text']
     return story_text
 
@@ -39,6 +39,4 @@ if uploaded_file is not None:
 
     st.text('Generating audio data...')
     audio_data = text2audio(story)
-
-    if st.button("Play Audio"):
-        st.audio(audio_data['audio'], format="audio/wav", start_time=0, sample_rate = audio_data['sampling_rate'])
+    st.audio(audio_data['audio'], format="audio/wav", start_time=0, sample_rate = audio_data['sampling_rate'])