Update app.py

app.py

@@ -15,10 +15,10 @@ def img2text(url):
         text = text.replace(word, "")
     return text.strip()
 
-# Function to generate a story from text using GPT-Neo
+# Function to generate a story from text using T5
 def text2story(text):
-    # Use the GPT-Neo model for storytelling
-    text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M")
+    # Use the T5 model for text generation
+    text_generator = pipeline("text2text-generation", model="t5-small")
     
     # Add a strong prompt to guide the model
     prompt = f"Write a short, happy, and fun story for kids aged 3-10. " \
@@ -41,7 +41,7 @@ def text2story(text):
 
 # Function to convert text to audio using gTTS
 def text2audio(story_text):
-    audio_file = "kids_playing_audio.wav"
+    audio_file = os.path.abspath("kids_playing_audio.wav")
     tts = gTTS(story_text, lang="en")
     tts.save(audio_file)
     return audio_file
@@ -65,7 +65,7 @@ if uploaded_file is not None:
     with open(uploaded_file.name, "wb") as file:
         file.write(bytes_data)
 
-    st.image(uploaded_file, caption="Your Uploaded Image", use_container_width=True)
+    st.image(uploaded_file, caption="Your Uploaded Image", use_column_width=True)
 
     # Stage 1: Image to Text
     st.text('🖼️ Processing image...')
@@ -84,7 +84,10 @@ if uploaded_file is not None:
 
     # Play button for audio
     if st.session_state.audio_file and st.button("🎧 Listen to the Story"):
-        st.audio(st.session_state.audio_file, format="audio/wav")
+        if os.path.exists(st.session_state.audio_file):
+            st.audio(st.session_state.audio_file, format="audio/wav")
+        else:
+            st.error("Audio file not found. Please generate the story again.")
 
     # Clean up the generated audio file
     if st.session_state.audio_file and os.path.exists(st.session_state.audio_file):