Update Dockerfile

Dockerfile

@@ -5,7 +5,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends gcc libc6-dev &
 
 WORKDIR /app
 
-# Install Python dependencies (added huggingface_hub which includes the CLI)
+# Install Python dependencies
 RUN pip install --no-cache-dir flask flask-cors Pillow litert-lm huggingface_hub
 
 # Create dummy Vulkan library to satisfy litert-lm on CPU
@@ -16,9 +16,8 @@ RUN printf 'void vkGetInstanceProcAddr(){}\nvoid vkCreateInstance(){}\nvoid vkDe
 # Create model directory
 RUN mkdir -p /app/models/gemma
 
-# --- KEY CHANGE: Download the model during the Docker build ---
-# This bakes the model into your Hugging Face Space image
-RUN huggingface-cli download litert-community/gemma-4-E2B-it-litert-lm gemma-4-E2B-it.litertlm --local-dir /app/models/gemma
+# --- KEY FIX: Use the new 'hf' command instead of 'huggingface-cli' ---
+RUN hf download litert-community/gemma-4-E2B-it-litert-lm gemma-4-E2B-it.litertlm --local-dir /app/models/gemma
 
 # Copy the server code
 COPY server.py .
@@ -31,7 +30,6 @@ USER user
 # Set environment variables
 ENV PORT=7860
 ENV LD_LIBRARY_PATH=/app
-# --- KEY CHANGE: Point the Python script directly to the baked-in model ---
 ENV GEMMA_MODEL_PATH=/app/models/gemma/gemma-4-E2B-it.litertlm
 
 EXPOSE 7860