Initial commit

Dockerfile

@@ -5,7 +5,7 @@ RUN useradd -m -u 1000 user
 
 # Set Hugging Face cache to user's writable directory
 ENV HF_HOME=/home/user/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
+ENV TRANSFORMERS_CACHE=/data/huggingface_cache
 
 # Create cache directory with proper permissions
 RUN mkdir -p ${HF_HOME} && chown -R user:user /home/user

api.py

@@ -54,7 +54,7 @@ def load_model():
     with MODEL_LOCK:
         if MODEL is None:
             print("⏳ Loading music generation model...")
-            MODEL = AutoModelForCausalLM.from_pretrained('stanford-crfm/music-small-800k',local_files_only=True, force_download=False)    # Prevent re-downloads
+            MODEL = AutoModelForCausalLM.from_pretrained('stanford-crfm/music-small-800k',local_files_only=False, force_download=False)    # Prevent re-downloads
             # Add .cuda() here if using GPU
             print("✅ Model loaded successfully!")
     return MODEL