fix

app.py

@@ -1,6 +1,6 @@
 
 import warnings
-import spaces#space
+import spaces
 warnings.filterwarnings("ignore", category=FutureWarning)
 import logging
 from argparse import ArgumentParser
@@ -27,6 +27,10 @@ from datetime import datetime
 
 log = logging.getLogger()
 
+device = "cpu"
+if torch.cuda.is_available():
+    device = "cuda"
+
 setup_eval_logging()
 
 OUTPUT_DIR = Path("./output/gradio")
@@ -335,9 +339,7 @@ with gr.Blocks(title="MeanAudio Generator", theme=theme, css=custom_css) as demo
                 list(all_model_cfg.keys()) if all_model_cfg else []
             )
             default_variant = (
-                "small_16k_mf"
-                if "small_16k_mf" in available_variants
-                else available_variants[0] if available_variants else ""
+                'meanaudio_mf'
             )
             variant = gr.Dropdown(
                 label="Model Variant",
@@ -406,5 +408,8 @@ with gr.Blocks(title="MeanAudio Generator", theme=theme, css=custom_css) as demo
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    parser = ArgumentParser()
+    parser.add_argument("--port", type=int, default=7861)
+    args = parser.parse_args()
+    demo.launch(server_port=args.port, allowed_paths=[OUTPUT_DIR.resolve()])