Update app.py

app.py

@@ -3,6 +3,12 @@ import torch
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Tokenizer
 import streamlit as st
 
+from audio_recorder_streamlit import audio_recorder
+
+audio_bytes = audio_recorder()
+if audio_bytes:
+    st.audio(audio_bytes, format="audio/wav")
+    
 #load pre-trained model and tokenizer
 tokenizer = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-base-960h")
 model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
@@ -23,8 +29,4 @@ transcriptions = tokenizer.decode(predicted_ids[0])
 
 print(transcriptions)
 
-
-
-
-
 st.write("hi")