Update app.py

app.py

@@ -6,7 +6,8 @@ from pydub import AudioSegment
 from asr import Transcribe
 
 
-def transcribe(audio_file, lang_id: str):
+def transcribe(audio_file, language: str):
+    language_dict = {"Amharic":"amh","Oromo":"orm","Somali":"som"}
     freq = 16000
     # Return the transcript.
     transcript = ""
@@ -19,11 +20,11 @@ def transcribe(audio_file, lang_id: str):
     if orig_freq != freq:
         waveform = AF.resample(waveform, orig_freq, freq)
 
-    return transcriber(waveform, lang_id), audio_file.name
+    return transcriber(waveform, langauge_dict[language]), audio_file.name
 
 if __name__ == "__main__":
     transcriber = Transcribe()
-    inputs = [gr.File(), gr.Dropdown(choices=["amh", "orm", "som"])]
+    inputs = [gr.File(), gr.Dropdown(choices=["Amharic", "Oromo", "Somali"])]
     outputs = [
         gr.Textbox(label="Transcript"),
         gr.Audio(label="Audio", type="filepath"),