Add Docker setup for Hugging Face Spaces deployment

Dockerfile

@@ -0,0 +1,16 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy source
+COPY src/ ./src/
+COPY checkpoints/ ./checkpoints/
+
+# Expose port Hugging Face expects
+EXPOSE 7860
+
+CMD ["python", "-m", "uvicorn", "src.inference:app", "--host", "0.0.0.0", "--port", "7860"]

requirements.txt

@@ -0,0 +1,7 @@
+fastapi
+uvicorn
+python-multipart
+nibabel
+numpy
+python-dotenv
+torch==2.5.1+cpu --extra-index-url https://download.pytorch.org/whl/cpu

src/inference.py

@@ -43,7 +43,7 @@ app = FastAPI(
 
 app.add_middleware(
     CORSMiddleware,
-    allow_origins=["http://localhost:5173", "http://localhost:3000"],
+    allow_origins=["*"],
     allow_methods=["*"],
     allow_headers=["*"],
 )