Update app.py

app.py

@@ -69,8 +69,8 @@ def ui():
             verbose=True,
             memory=ConversationBufferWindowMemory(k=2)
         )
-        good_morining_audio,sample_rate=librosa.load('./good-morning.mp3')
-        hi_audio,sample_rate=librosa.load('./good-morning-sir.mp3')
+        good_morining_audio,sample_rate1=librosa.load('./good-morning.mp3')
+        hi_audio,sample_rate2=librosa.load('./good-morning-sir.mp3')
         if 'history' not in st.session_state:
             st.session_state['history'] = []
     
@@ -122,8 +122,12 @@ def ui():
             if 'generated' in st.session_state and st.session_state['generated']:
                 with response_container:
                     for i in range(len(st.session_state['generated'])):
-                        st.audio(st.session_state["past"][i],format='audio/wav')
-                        st.audio(st.session_state["generated"][i],format='audio/wav')
+                        if i==0:
+                            st.audio(st.session_state["past"][i],sample_rate=sample_rate1,format='audio/wav')
+                            st.audio(st.session_state["generated"][i],sample_rate=sample_rate2,format='audio/wav')
+                        else:
+                            st.audio(st.session_state["past"][i],format='audio/wav')
+                            st.audio(st.session_state["generated"][i],format='audio/wav')
 
 if __name__ == '__main__':
     ui()