Rollback to MT3 and adding gradio box

app.py

@@ -25,7 +25,7 @@ os.putenv("QT_QPA_PLATFORM", "offscreen")
 os.putenv("XDG_RUNTIME_DIR", environment.Environment().getRootTempDir())
 
 # Start inference model
-inference_model = InferenceModel("/home/user/app/checkpoints/ismir2021/", "ismir2021")
+inference_model = InferenceModel("/home/user/app/checkpoints/mt3/", "mt3")
 current_model = "ismir2021"
 
 def change_model(model):
@@ -93,35 +93,38 @@ with demo:
                 + "</h1>")
     gr.Markdown(description)
     with gr.Box():
-        model_label = """
-        What kind of model you want to use? 
-        The ismir2021 model transcribes piano only, with note velocities. 
-        The mt3 model transcribes multiple simultaneous instruments, but without velocities.
-        """
-        model = gr.Radio(
-            ["mt3", "ismir2021"], 
-            label=model_label, 
-            value="ismir2021"
-        )
-        model.change(fn=change_model, inputs=model, outputs=[])
-
-        with gr.Row():
-            link = gr.Textbox(label="YouTube Link")
-        with gr.Row():
-            preview_btn = gr.Button("Preview")
-        with gr.Row().style(mobile_collapse=False, equal_height=True):
-            title = gr.Label(label="Video Title", placeholder="Title")
-            img = gr.Image(label="Thumbnail")
-        with gr.Row():
-            yt_audio = gr.Audio()
-            yt_audio_path = gr.Textbox(visible=False)
-
-        preview_btn.click(fn=populate_metadata,
-                        inputs=[link],
-                        outputs=[img, title, yt_audio, yt_audio_path])
-        
-        with gr.Row():
-            btn = gr.Button("Transcribe music")
+        with gr.Box():
+            model_label = """
+            What kind of model you want to use? 
+            The ismir2021 model transcribes piano only, with note velocities. 
+            The mt3 model transcribes multiple simultaneous instruments, but without velocities.
+            """
+            model = gr.Radio(
+                ["mt3", "ismir2021"], 
+                label=model_label, 
+                value="mt3"
+            )
+            model.change(fn=change_model, inputs=model, outputs=[])
+
+            with gr.Row():
+                link = gr.Textbox(label="YouTube Link")
+            with gr.Row():
+                preview_btn = gr.Button("Preview")
+                
+        with gr.Box():
+            with gr.Row().style(mobile_collapse=False, equal_height=True):
+                title = gr.Label(label="Video Title", placeholder="Title")
+                img = gr.Image(label="Thumbnail")
+            with gr.Row():
+                yt_audio = gr.Audio()
+                yt_audio_path = gr.Textbox(visible=False)
+
+            preview_btn.click(fn=populate_metadata,
+                            inputs=[link],
+                            outputs=[img, title, yt_audio, yt_audio_path])
+            
+            with gr.Row():
+                btn = gr.Button("Transcribe music")
         
         with gr.Row():
             midi_file = gr.File()