feat: initialize git repository and update project configuration and dependencies

Dockerfile

@@ -0,0 +1,42 @@
+FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
+
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    git ffmpeg libsm6 libxext6 cmake rsync libgl1 curl \
+    && rm -rf /var/lib/apt/lists/*
+
+# Install uv for fast python package installations
+RUN pip install uv
+
+# Create non-root user required by HuggingFace Spaces
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH \
+    PYTHONWARNINGS=ignore \
+    GRADIO_SERVER_NAME="0.0.0.0" \
+    GRADIO_SERVER_PORT="7860"
+
+WORKDIR $HOME/app
+
+# Copy requirements and install base dependencies via uv
+COPY --chown=user requirements.txt $HOME/app/
+RUN uv pip install --system --upgrade pip
+RUN uv pip install --system -r requirements.txt
+
+# Specify CUDA architectures to compile for various GPUs commonly found on HF Spaces 
+# 7.5: T4, 8.0: A100, 8.6: A10G/RTX3090, 8.9: L4, 9.0: H100
+ENV TORCH_CUDA_ARCH_LIST="7.5 8.0 8.6 8.9 9.0+PTX"
+
+# Install causal-conv1d and mamba-ssm requiring compilation 
+# Since this uses the `devel` image, `nvcc` is available safely!
+RUN uv pip install --system causal-conv1d==1.5.0.post8 --no-build-isolation
+RUN uv pip install --system mamba-ssm==2.2.4 --no-build-isolation
+
+# Copy the rest of the application code
+COPY --chown=user . $HOME/app/
+
+# Run the Gradio app
+CMD ["python", "app.py"]

README.md

@@ -3,7 +3,7 @@ title: MambaEye
 emoji: 👁️
 colorFrom: blue
 colorTo: green
-sdk: gradio
+sdk: docker
 sdk_version: 6.10.0
 app_file: app.py
 pinned: false

requirements.txt

@@ -6,5 +6,3 @@ torchvision==0.21.0
 lightning==2.6.1
 huggingface_hub
 omegaconf==2.3.0
-mamba-ssm==2.2.4
-causal-conv1d==1.5.0.post8