Update app.py

app.py

@@ -24,7 +24,16 @@ with st.sidebar:
     # ToppSide = st.slider("Top-p (nucleus sampling)", min_value=0.0, max_value=1.0, value=0.6, step=0.05)
     # RepetitionpenaltySide = st.slider("Repetition penalty", min_value=0.0, max_value=2.0, value=1.2, step=0.05)
 
+whisper_client = Client("https://sanchit-gandhi-whisper-large-v2.hf.space/")
 
+
+def transcribe(wav_path):
+    
+    return whisper_client.predict(
+				wav_path,	# str (filepath or URL to file) in 'inputs' Audio component
+				"transcribe",	# str in 'Task' Radio component
+				api_name="/predict"
+    )
     
 # Prediction function
 def predict(message, system_prompt='', temperature=0.7, max_new_tokens=4096,Topp=0.5,Repetitionpenalty=1.2):
@@ -52,8 +61,18 @@ st.write(DESCRIPTION)
 textinput = st.chat_input("Ask LLama-2-70b anything...")
 wav_audio_data = st_audiorec()
 
-
+if wav_audio_data != None:
+    # save audio
+    with open("audio.wav", "wb") as f:
+        f.write(wav_audio_data)
+    # transcribe audio
+    response = transcribe("audio.wav")
     
+    with st.chat_message("assistant", avatar='🦙'):
+        st.markdown(response)
+    # Add assistant response to chat history
+    st.session_state.messages.append({"role": "assistant", "content": response})
+
 
 
 if "messages" not in st.session_state:
@@ -76,8 +95,4 @@ if prompt := textinput:
     with st.chat_message("assistant", avatar='🦙'):
         st.markdown(response)
     # Add assistant response to chat history
-    st.session_state.messages.append({"role": "assistant", "content": response})
-    
-
-    
-    
+    st.session_state.messages.append({"role": "assistant", "content": response})