lpw commited on
Commit
576543f
1 Parent(s): b2f0be5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -21,10 +21,12 @@ for src in SRC_LIST:
21
  MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
22
 
23
  examples = []
 
 
24
 
25
  # io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}", api_key=st.secrets["api_key"]) for model in MODEL_LIST}
26
  # io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}") for model in MODEL_LIST}
27
- io_dict = {model: load_model_ensemble_and_task_from_hf_hub(f"facebook/{model}")[0] for model in MODEL_LIST}
28
 
29
  def inference(audio, model):
30
  out_audio = io_dict[model](audio)
 
21
  MODEL_LIST.append(f"textless_sm_{src}_{tgt}")
22
 
23
  examples = []
24
+ MODEL_LIST = ['textless_sm_es_en', 'textless_sm_es_fr', 'textless_sm_et_en']
25
+ print(MODEL_LIST)
26
 
27
  # io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}", api_key=st.secrets["api_key"]) for model in MODEL_LIST}
28
  # io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}") for model in MODEL_LIST}
29
+ io_dict = {model: load_model_ensemble_and_task_from_hf_hub(f"facebook/{model}") for model in MODEL_LIST}
30
 
31
  def inference(audio, model):
32
  out_audio = io_dict[model](audio)