Change way to define inference model

app.py

@@ -24,20 +24,12 @@ us["musescoreDirectPNGPath"] = "/usr/bin/mscore3"
 os.putenv("QT_QPA_PLATFORM", "offscreen")
 os.putenv("XDG_RUNTIME_DIR", environment.Environment().getRootTempDir())
 
-# Start inference model
-inference_model = InferenceModel("/home/user/app/checkpoints/mt3/", "mt3")
-current_model = "ismir2021"
-
-def change_model(model):
-    global current_model
-    global inference_model
-    print("Inferece model", inference_model)
-    print("Current model", current_model)
-    checkpoint_path = f"/home/user/app/checkpoints/{model}/"
-    if model == current_model:
-        return
+def load_model(model=str):
+    checkpoint_path = f'/content/checkpoints/{model}/'
+    # Start inference model
     inference_model = InferenceModel(checkpoint_path, model)
-    current_model = model
+    return inference_model
+    
 
 # Credits https://huggingface.co/spaces/rajesh1729/youtube-video-transcription-with-whisper
 def get_audio(url):
@@ -56,12 +48,14 @@ def populate_metadata(link):
     audio = get_audio(link)
     return yt.thumbnail_url, yt.title, audio, audio
 
-def inference(yt_audio_path):
+def inference(yt_audio_path, model):
 
     with open(yt_audio_path, 'rb') as fd:
       contents = fd.read()
     
     audio = upload_audio(contents,sample_rate=SAMPLE_RATE)
+    
+    inference_model = load_model(model)
 
     est_ns = inference_model(audio)
     
@@ -104,7 +98,6 @@ with demo:
                 label=model_label, 
                 value="mt3"
             )
-            model.change(fn=change_model, inputs=model, outputs=[])
 
             with gr.Row():
                 link = gr.Textbox(label="YouTube Link")
@@ -120,7 +113,7 @@ with demo:
                 yt_audio_path = gr.Textbox(visible=False)
 
             preview_btn.click(fn=populate_metadata,
-                            inputs=[link],
+                            inputs=[link, model],
                             outputs=[img, title, yt_audio, yt_audio_path])
             
             with gr.Row():