Upload 2 files

Dockerfile

@@ -10,6 +10,9 @@ RUN apt-get update && apt-get install -y \
     ffmpeg \
     && rm -rf /var/lib/apt/lists/*
 
+# Create cache directory with proper permissions
+RUN mkdir -p /app/.cache && chmod 777 /app/.cache
+
 # Copy requirements and install Python dependencies
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
@@ -23,6 +26,9 @@ EXPOSE 7860
 # Set environment variables
 ENV PYTHONUNBUFFERED=1
 ENV PORT=7860
+ENV HF_HOME=/app/.cache
+ENV HF_DATASETS_CACHE=/app/.cache
+ENV TRANSFORMERS_CACHE=/app/.cache
 
 # Run the FastAPI application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] 

app.py

@@ -31,11 +31,22 @@ async def lifespan(app: FastAPI):
     try:
         from sonics import HFAudioClassifier
         print("🔄 Loading Madverse Music AI model...")
-        model = HFAudioClassifier.from_pretrained("awsaf49/sonics-spectttra-alpha-120s")
+        
+        # Set cache directory to a writable location
+        cache_dir = "/app/.cache" if os.path.exists("/app") else "./cache"
+        os.makedirs(cache_dir, exist_ok=True)
+        
+        # Load model with explicit cache directory
+        model = HFAudioClassifier.from_pretrained(
+            "awsaf49/sonics-spectttra-alpha-120s",
+            cache_dir=cache_dir
+        )
         model.eval()
         print("✅ Model loaded successfully!")
     except Exception as e:
         print(f"❌ Failed to load model: {e}")
+        import traceback
+        traceback.print_exc()
         raise
     
     yield