Upload 3 files

Dockerfile

@@ -1,22 +1,43 @@
 FROM pytorch/pytorch:2.6.0-cuda12.4-cudnn9-devel
 
+# 1. System prerequisites setup as Root
+USER root
 WORKDIR /app
 
 RUN apt-get update && apt-get install -y \
     git \
     && rm -rf /var/lib/apt/lists/*
 
-RUN git clone --recurse-submodules https://github.com/ZHZisZZ/dllm.git /tmp/dllm && \
-    pip install --no-cache-dir -e /tmp/dllm && \
-    pip install --no-cache-dir -e /tmp/dllm/lm-evaluation-harness
+# Clone framework natively into an application staging directory
+RUN git clone --recurse-submodules https://github.com /app/dllm-source
 
-RUN pip install --no-cache-dir flask
+# Install the library globally so it compiles securely across all user sessions
+RUN pip install --no-cache-dir /app/dllm-source && \
+    pip install --no-cache-dir /app/dllm-source/lm-evaluation-harness
 
+# Install remaining Python package prerequisites 
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy deployment script 
 COPY app.py .
 
+# 2. Hugging Face Spaces Sandbox Compliance 
+# Set up a generic user mapping to UID 1000
+RUN useradd -m -u 1000 user && \
+    chown -R user:user /app
+
+USER user
+ENV HOME=/home/user
+ENV PATH=/home/user/.local/bin:$PATH
+
+# Direct Hugging Face cache downloads into a writeable workspace location
+ENV HF_HOME=/home/user/.cache/huggingface
+
 ENV MODEL_NAME=dllm-hub/Qwen3-0.6B-diffusion-bd3lm-v0.1
 ENV PORT=7860
 
 EXPOSE 7860
 
 CMD ["python", "app.py"]
+

app.py

@@ -1,4 +1,9 @@
 import os
+import sys
+
+# Resolves relative module routes for trust_remote_code executors
+sys.path.append(os.path.abspath("/app/dllm-source"))
+
 import math
 import copy
 import json

requirements.txt

@@ -3,4 +3,3 @@ torch
 transformers
 accelerate
 numpy<2.0
-git+https://github.com/ZHZisZZ/dllm.git