Update app.py

app.py

@@ -1,22 +1,26 @@
 import gradio as gr
-from audiocraft.data.audio import audio_write
-import audiocraft.models
 import numpy as np
 import pandas as pd
 import os
 import torch
 
+# Change the directory
+os.chdir('./audiocraft')
+from audiocraft.data.audio import audio_write
+import audiocraft.models
+
+
 # download models
 huggingface_hub.hf_hub_download(
     repo_id='Cyan0731/MusiConGen',
     filename='compression_state_dict.bin',
-    local_dir='./audiocraft/ckpt/musicongen'
+    local_dir='./ckpt/musicongen'
 )
 
 huggingface_hub.hf_hub_download(
     repo_id='Cyan0731/MusiConGen',
     filename='state_dict.bin',
-    local_dir='./audiocraft/ckpt/musicongen'
+    local_dir='./ckpt/musicongen'
 )
 
 def print_directory_contents(path):
@@ -41,7 +45,7 @@ def infer(text):
     
     
     # load your model
-    musicgen = audiocraft.models.MusicGen.get_pretrained('./audiocraft/ckpt/musicongen') ### change this path
+    musicgen = audiocraft.models.MusicGen.get_pretrained('./ckpt/musicongen') ### change this path
     musicgen.set_generation_params(duration=duration, extend_stride=duration//2, top_k = 250)