add token env in dockerfile

SukhmeetSingh2002 · Oct 1, 2023 · 1da8844 · 1da8844
1 parent b71da5c
commit 1da8844
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -16,9 +16,10 @@ COPY . .
 # Expose port 5000 for the Flask web application
 EXPOSE 5000
 
-# Set the environment variables for Flask
+# Set the environment variables for Flask and HF_TOKEN
 ENV FLASK_APP=app.py
 ENV FLASK_ENV=production
+ENV HF_TOKEN=${HF_TOKEN}
 
 # Run the command to start the Flask web application
 # CMD ["flask", "run", "--host=0.0.0.0"]