add ability to chose a model

app.py

@@ -3,13 +3,20 @@ import whisper
 import tempfile
 from pydub import AudioSegment
 
-# Load the Whisper model
-model = whisper.load_model("base")
+# Define available models
+available_models = ["tiny", "base", "small", "medium", "large"]
 
-st.title("Voice Recognition App using Whisper")
+st.title("Voice Recognition App")
 
-st.write("Upload an audio file and the Whisper model will transcribe it to text.")
+st.write("Upload an audio file and choose a Whisper model to transcribe it to text.")
 
+# Model selection dropdown
+model_choice = st.selectbox("Choose a Whisper model", available_models)
+
+# Load the selected Whisper model
+st.write(f"Loading {model_choice} model...")
+model = whisper.load_model(model_choice)
+st.write(f"{model_choice} model loaded successfully.")
 # File uploader for audio file
 uploaded_file = st.file_uploader("Choose an audio file", type=["wav", "mp3", "m4a"])