Update handler.py

handler.py

@@ -11,7 +11,6 @@ logging.basicConfig(level=logging.ERROR)
 # Configure logging
 logging.basicConfig(level=logging.WARNING)
 
-import streamlit as st
 import pygame
 
 class EndpointHandler():
@@ -33,10 +32,10 @@ class EndpointHandler():
 
         # Save the audio to a file
         sf.write("StoryAudio.wav", outputs["waveform"][0].numpy(), self.model.config.sampling_rate)
-        st.audio("StoryAudio.wav")
-        # pygame.mixer.init()
-        # pygame.mixer.music.load("StoryAudio.wav")
-        # pygame.mixer.music.play()
+        #st.audio("StoryAudio.wav")
+        pygame.mixer.init()
+        pygame.mixer.music.load("StoryAudio.wav")
+        pygame.mixer.music.play()
         #return 'StoryAudio.wav'
         #return {"audio_file_path": 'StoryAudio.wav'}
         # with open("StoryAudio.wav", "rb") as f: