Update app.py

app.py

@@ -3,6 +3,7 @@ import streamlit as st
 from transformers import pipeline
 import soundfile as sf
 import numpy as np
+import tempfile
 
 # function part
 # img2text
@@ -13,7 +14,7 @@ def img2text(url):
 
 # text2story
 def text2story(text):
-    story_text_model = pipeline("text-generation", model="meta-llama/Llama-3.1-8B")
+    story_text_model = pipeline("text-generation", model="google/gemma-2-9b-it")
     story = story_text_model(text, max_length=150)[0]['generated_text']
     return story
 
@@ -22,8 +23,8 @@ def text2audio(story_text):
     tts_model = pipeline("text-to-speech", model="tts_models/en/ljspeech/tacotron2")
     audio_data = tts_model(story_text)
     
-    # Save audio to a file
-    audio_filename = "story_audio.wav"
+    # Save audio to a temporary file
+    audio_filename = tempfile.mktemp(suffix=".wav")
     sf.write(audio_filename, audio_data['audio'], audio_data['sampling_rate'])
     
     return audio_filename