Initial commit

Dockerfile

@@ -1,31 +1,25 @@
 FROM python:3.9-slim
 
-# Create dedicated user with home directory
+# Create user with UID 1000
 RUN useradd -m -u 1000 user
 
-# Set Hugging Face cache to user's writable directory
+# Use writable cache location in user's home directory
 ENV HF_HOME=/home/user/.cache/huggingface
-ENV TRANSFORMERS_CACHE=/data/huggingface_cache
+ENV TRANSFORMERS_CACHE=/home/user/.cache/huggingface
 
 # Create cache directory with proper permissions
 RUN mkdir -p ${HF_HOME} && chown -R user:user /home/user
 
-# Set working directory (app will live here)
 WORKDIR /app
 
-# Install dependencies as root
+# Install dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt gunicorn
 
-# Copy app files (maintain ownership)
+# Copy application files
 COPY --chown=user:user . .
 
-RUN rm -rf /root/.cache/pip
-
-
-# Switch to non-root user
 USER user
 
-
 EXPOSE 7860
-CMD ["gunicorn", "--workers", "1", "--timeout", "120", "--bind", "0.0.0.0:7860", "api:app"]
+CMD ["gunicorn", "--workers", "1", "--timeout", "300", "--bind", "0.0.0.0:7860", "api:app"]

api.py

@@ -50,14 +50,25 @@ def midi_to_musicxml(midi_file_path):
         raise
 
 def load_model():
-    global 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=False, force_download=False)    # Prevent re-downloads
-            # Add .cuda() here if using GPU
-            print("✅ Model loaded successfully!")
-    return MODEL
+    cache_dir = os.environ.get('HF_HOME', '/home/user/.cache/huggingface')
+    print(f"Using cache directory: {cache_dir}")
+    
+    # Verify permissions
+    try:
+        test_file = os.path.join(cache_dir, "test.txt")
+        with open(test_file, "w") as f:
+            f.write("test")
+        print("✅ Cache directory is writable")
+    except Exception as e:
+        print(f"❌ Cache directory not writable: {e}")
+
+    # Load model
+    return AutoModelForCausalLM.from_pretrained(
+        'stanford-crfm/music-small-800k',
+        cache_dir=cache_dir,
+        local_files_only=False,
+        force_download=False
+    )
 
 # Model loading setup
 MODEL = None