Update Dockerfile

Dockerfile

@@ -1,29 +1,28 @@
 # syntax=docker/dockerfile:1.4
 FROM ubuntu:24.04
-
 ENV DEBIAN_FRONTEND=noninteractive
-RUN apt-get update && apt-get install -y \
-    python3 python3-venv python3-pip git wget \
-  && rm -rf /var/lib/apt/lists/*
+
+RUN apt-get update && apt-get install -y python3-venv python3-pip git wget && \
+    rm -rf /var/lib/apt/lists/*
 
 WORKDIR /workspace
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git
 WORKDIR /workspace/ComfyUI
 
-RUN python3 -m venv .venv && \
-    . .venv/bin/activate && \
-    pip install --upgrade pip && \
-    pip install xformers!=0.0.18 && \
+RUN python3 -m venv .venv && . .venv/bin/activate && \
+    pip install --upgrade pip && pip install xformers!=0.0.18 && \
     pip install -r requirements.txt
 
-RUN --mount=type=secret,id=hf_token,required=true \
+RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true \
     mkdir -p models/unet models/vae && \
-    wget --header="Authorization: Bearer $(cat /run/secrets/hf_token)" \
+    hf=$(cat /run/secrets/HF_TOKEN) && \
+    wget --header="Authorization: Bearer $hf" \
       https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/flux1-schnell.safetensors \
       -O models/unet/flux1-schnell.safetensors && \
-    wget --header="Authorization: Bearer $(cat /run/secrets/hf_token)" \
+    wget --header="Authorization: Bearer $hf" \
       https://huggingface.co/black-forest-labs/FLUX.1-schnell/resolve/main/ae.safetensors \
       -O models/vae/ae.safetensors
 
 EXPOSE 7860
-CMD ["/bin/bash", "-lc", "source .venv/bin/activate && python3 main.py --host 0.0.0.0 --port 7860 --cpu"]
+
+CMD ["bash", "-lc", "source .venv/bin/activate && python3 main.py --host 0.0.0.0 --port 7860 --cpu"]