lomit commited on
Commit
2f60ebd
·
1 Parent(s): 71ddb11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -128,14 +128,16 @@ if file_data is not None:
128
  if selected_model_size in model_option:
129
  model = load_model(selected_model_size)
130
 
131
- lang_button = st.button("What is langage")
132
  if lang_button:
133
- probs = get_langage_cls(audio_arr=audio_arr, model=model)
 
134
  st.write(f"Detected language: {max(probs, key=probs.get)}")
135
 
136
  task_option = ["transcribe", "translate"]
137
  translate_task = st.selectbox("What is your task", ["None"] + task_option)
138
 
139
  if translate_task != "None":
140
- result = transcribe(audio=audio_arr, model=model, task=translate_task)
 
141
  st.write(result)
 
128
  if selected_model_size in model_option:
129
  model = load_model(selected_model_size)
130
 
131
+ lang_button = st.button("What is language")
132
  if lang_button:
133
+ with st.spinner('Detecting language...'):
134
+ probs = get_langage_cls(audio_arr=audio_arr, model=model)
135
  st.write(f"Detected language: {max(probs, key=probs.get)}")
136
 
137
  task_option = ["transcribe", "translate"]
138
  translate_task = st.selectbox("What is your task", ["None"] + task_option)
139
 
140
  if translate_task != "None":
141
+ with st.spinner('In progress...'):
142
+ result = transcribe(audio=audio_arr, model=model, task=translate_task)
143
  st.write(result)