Upload Dockerfile

Dockerfile

@@ -31,6 +31,7 @@ ENV NODE_ENV=production \
     NEXT_PUBLIC_DB_MODE=browser \
     NEXT_TELEMETRY_DISABLED=1 \
     PORT=7860 \
+    DATA_ROOT=/data \
     DATASETS_FOLDER=/data/datasets \
     TRAINING_FOLDER=/data/output \
     HF_JOBS_DEFAULT_HARDWARE=a100-large \
@@ -49,11 +50,11 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     && ln -s /root/.local/bin/uv /usr/local/bin/uv \
     && apt-get clean && rm -rf /var/lib/apt/lists/*
 
-RUN mkdir -p /data/datasets /data/output && chmod -R 777 /data
+RUN mkdir -p /data/datasets /data/output /data/images && chmod -R 777 /data
 
 COPY --from=build /app /app
 COPY --from=deps /app/ui/node_modules ./node_modules
 
 EXPOSE 7860
 ENTRYPOINT ["tini", "--"]
-CMD ["npm", "run", "start", "--", "--hostname", "0.0.0.0", "--port", "7860"]
+CMD ["npm", "run", "start", "--", "--hostname", "0.0.0.0", "--port", "7860"]