Update app.py

app.py

@@ -1,6 +1,7 @@
 import torch
 from transformers import Wav2Vec2Processor, Wav2Vec2ForCTC
 import soundfile as sf
+import streamlit as st
 
 # Load the processor and model
 processor = Wav2Vec2Processor.from_pretrained("openbmb/MiniCPM-o-2_6")
@@ -23,7 +24,13 @@ def transcribe_audio(file_path):
     
     return transcription[0]
 
-if __name__ == "__main__":
-    audio_file_path = "CAR0005.mp3"
-    transcription = transcribe_audio(audio_file_path)
-    print("Transcription:", transcription)
+uploaded_file = st.file_uploader("Upload an audio", type=["mp3", "wav"])
+
+if uploaded_file is not None:
+     transcription = transcribe_audio(uploaded_file)
+     st.write(transcription)
+    
+# if __name__ == "__main__":
+#     audio_file_path = "CAR0005.mp3"
+#     transcription = transcribe_audio(audio_file_path)
+#     print("Transcription:", transcription)