update model, processor

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
-from transformers import AutoModelForSpeechSeq2Seq, Wav2Vec2Processor
+# from transformers import AutoModelForSpeechSeq2Seq, Wav2Vec2Processor
+from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
 import torch
 import tempfile
 from pydub import AudioSegment
@@ -25,11 +26,16 @@ model_choice = st.selectbox("Choose a SpeechSeq2Seq model", available_models)
 # Load the selected model and processor
 @st.cache_resource
 def load_model_and_processor(model_name):
+    # model = AutoModelForSpeechSeq2Seq.from_pretrained(model_name)
+    # processor = Wav2Vec2Processor.from_pretrained(model_name)
     model = AutoModelForSpeechSeq2Seq.from_pretrained(model_name)
-    processor = Wav2Vec2Processor.from_pretrained(model_name)
+    processor = AutoProcessor.from_pretrained(model_name)
     return model, processor
 
 
+processor = AutoProcessor.from_pretrained("Yehor/whisper-small-ukrainian")
+model = AutoModelForSpeechSeq2Seq.from_pretrained("Yehor/whisper-small-ukrainian")
+
 st.write(f"Loading {model_choice} model...")
 model, processor = load_model_and_processor(model_choice)
 st.write(f"{model_choice} model loaded successfully.")