Update app.py

app.py

@@ -1,13 +1,17 @@
 import torch
-from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
+#from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
+from transformers import Speech2Text2Processor, SpeechEncoderDecoderModel
 import streamlit as st
 from audio_recorder_streamlit import audio_recorder
 import numpy as np
 
 # Function to transcribe audio to text
 def transcribe_audio(audio_bytes):
-    processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
-    model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
+    #processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
+    processor = Speech2Text2Processor.from_pretrained("facebook/s2t-wav2vec2-large-en-de")
+    model = SpeechEncoderDecoderModel.from_pretrained("facebook/s2t-wav2vec2-large-en-de")
+    #model = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-base-960h")
+    
 
     # Convert bytes to numpy array
     audio_array = np.frombuffer(audio_bytes, dtype=np.int16)