Update Dockerfile

Dockerfile

@@ -13,6 +13,15 @@ RUN pip install -r requirements.txt
 # Step 5: Expose the necessary port
 EXPOSE 7860
 
+# Expose the secret HUGGINGFACE_TOKEN at buildtime and use its value as the Hugging Face token
+RUN --mount=type=secret,id=HUGGINGFACE_TOKEN,mode=0444,required=true \
+    echo "HUGGINGFACE_TOKEN=$(cat /run/secrets/HUGGINGFACE_TOKEN)" > .env
+
+# Install huggingface-cli
+RUN pip install huggingface_hub
+
+# Login to Hugging Face using the token from the environment variable
+
 # Create a non-root user and grant read-write permissions
 RUN useradd -ms /bin/bash myuser
 RUN chown -R myuser:myuser /app
@@ -23,5 +32,7 @@ RUN chmod -R 777 /app
 # Switch to the non-root user
 USER myuser
 
+RUN huggingface-cli login --token $HUGGINGFACE_TOKEN
+
 # Step 6: Define the entry point command
 CMD ["flask", "run", "--host=0.0.0.0", "--port=7860"]