Update app.py

app.py

@@ -2,11 +2,22 @@ import gradio as gr
 import torch
 import torchaudio
 from torchaudio.transforms import Resample
+import os
 
-# Load MusicGen model
-device = "cuda" if torch.cuda.is_available() else "cpu"
-model = torch.hub.load("facebookresearch/audiocraft", "musicgen", source="github")
-model.to(device)
+# Clone and load MusicGen model
+def load_musicgen_model():
+    repo_path = "./audiocraft"  # Local path to the cloned repo
+    if not os.path.exists(repo_path):
+        os.system("git clone https://github.com/facebookresearch/audiocraft.git")
+    if repo_path not in os.sys.path:
+        os.sys.path.append(repo_path)
+
+    from audiocraft.models import MusicGen
+    model = MusicGen.get_pretrained("small")
+    model.set_device("cuda" if torch.cuda.is_available() else "cpu")
+    return model
+
+model = load_musicgen_model()
 
 def generate_music(prompt, duration=10):
     try: