Update demos/musicgen_app.py

demos/musicgen_app.py

@@ -203,8 +203,8 @@ def predict_batched(texts, melodies):
 
 
 def predict_full(
-    model,
-    model_path,
+    # model,
+    # model_path,
     decoder,
     text,
     melody,
@@ -219,16 +219,16 @@ def predict_full(
     global USE_DIFFUSION
     INTERRUPTING = False
     progress(0, desc="Loading model...")
-    model_path = model_path.strip()
-    if model_path:
-        if not Path(model_path).exists():
-            raise gr.Error(f"Model path {model_path} doesn't exist.")
-        if not Path(model_path).is_dir():
-            raise gr.Error(
-                f"Model path {model_path} must be a folder containing "
-                "state_dict.bin and compression_state_dict_.bin."
-            )
-        model = model_path
+    # model_path = model_path.strip()
+    # if model_path:
+    #     if not Path(model_path).exists():
+    #         raise gr.Error(f"Model path {model_path} doesn't exist.")
+    #     if not Path(model_path).is_dir():
+    #         raise gr.Error(
+    #             f"Model path {model_path} must be a folder containing "
+    #             "state_dict.bin and compression_state_dict_.bin."
+    #         )
+    #     model = model_path
     if temperature < 0:
         raise gr.Error("Temperature must be >= 0.")
     if topk < 0:
@@ -243,7 +243,8 @@ def predict_full(
         load_diffusion()
     else:
         USE_DIFFUSION = False
-    load_model(model)
+    load_model()
+    # load_model(model)
 
     max_generated = 0
 
@@ -377,8 +378,8 @@ def ui_full(launch_kwargs):
         ).then(
             predict_full,
             inputs=[
-                model,
-                model_path,
+                # model,
+                # model_path,
                 decoder,
                 text,
                 melody,