Update app.py

app.py

@@ -2,7 +2,9 @@ import os
 os.system("pip install gradio==3.3")
 import gradio as gr
 import numpy as np
-# import streamlit as st
+import streamlit as st
+
+from fairseq.checkpoint_utils import load_model_ensemble_and_task_from_hf_hub
 
 title = "SpeechMatrix Speech-to-speech Translation"
 
@@ -21,7 +23,8 @@ for src in SRC_LIST:
 examples = []
 
 # io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}", api_key=st.secrets["api_key"]) for model in MODEL_LIST}
-io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}") for model in MODEL_LIST}
+# io_dict = {model: gr.Interface.load(f"huggingface/facebook/{model}") for model in MODEL_LIST}
+io_dict = {model: load_model_ensemble_and_task_from_hf_hub(f"facebook/{model}") for model in MODEL_LIST}
    
 def inference(audio, model):
     out_audio = io_dict[model](audio)