try: loading model inside inference

app.py

@@ -2,27 +2,17 @@ import os
 import gradio as gr
 import numpy as np
 
-
-io1 = gr.load("huggingface/facebook/xm_transformer_s2ut_en-hk")
-io2 = gr.load("huggingface/facebook/xm_transformer_s2ut_hk-en")
-io3 = gr.load("huggingface/facebook/xm_transformer_unity_en-hk")
-io4 = gr.load("huggingface/facebook/xm_transformer_unity_hk-en")
-   
 def inference(audio, model):
-    print(f"[DEBUG] audio: {audio}")
-    print(f"[DEBUG] audio type: {type(audio)}")
-    print(f"[DEBUG] model: {model}")
-    out_audio = None
+    io = None
     if model == "xm_transformer_s2ut_en-hk":
-        out_audio = io1(audio)
+        io = gr.load("huggingface/facebook/xm_transformer_s2ut_en-hk")
     elif model == "xm_transformer_s2ut_hk-en":
-        out_audio = io2(audio)
+        io = gr.load("huggingface/facebook/xm_transformer_s2ut_hk-en")
     elif model == "xm_transformer_unity_en-hk":
-        out_audio = io3(audio)
+        io = gr.load("huggingface/facebook/xm_transformer_unity_en-hk")
     else:
-        out_audio = io4(audio)
-    print(f"[DEBUG] out: {out_audio}")
-    print(f"[DEBUG] out type: {type(out_audio)}")
+        io = gr.load("huggingface/facebook/xm_transformer_unity_hk-en")
+    out_audio = io(audio)
     return out_audio