chose from list, increase sampling rate

app.py

@@ -2,20 +2,26 @@ import streamlit as st
 import librosa
 from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq
 
+# Define available models
+available_models = ["Yehor/whisper-small-ukrainian"]
 
+st.title("Voice Recognition App")
 
-uploaded_file = st.file_uploader("Choose file", type=["wav", "mp3"])
+# Model selection dropdown
+model_choice = st.selectbox("Choose a model", available_models)
+
+processor = AutoProcessor.from_pretrained(model_choice)
 
-processor = AutoProcessor.from_pretrained("Yehor/whisper-small-ukrainian")
+model = AutoModelForSpeechSeq2Seq.from_pretrained(model_choice)
 
-model = AutoModelForSpeechSeq2Seq.from_pretrained("Yehor/whisper-small-ukrainian")
+uploaded_file = st.file_uploader("Choose file", type=["wav", "mp3"])
 
 def map_to_pred(file_path):
     # load audio file
     audio, _ = librosa.load(file_path)
 
     # preprocess audio and generate standard
-    input_features = processor([audio], return_tensors="pt", sampling_rate=16_000).input_features
+    input_features = processor([audio], return_tensors="pt", sampling_rate=32_000).input_features
     generated_ids = model.generate(inputs=input_features)
     transcription = processor.batch_decode(generated_ids, normalize=True, skip_special_tokens=True)
     text = processor.tokenizer._normalize(transcription[0])