Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -19,11 +19,14 @@ ENV HOME=/home/user \
|
|
| 19 |
GRADIO_NUM_PORTS=1 \
|
| 20 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 21 |
GRADIO_THEME=huggingface \
|
| 22 |
-
SYSTEM=spaces
|
|
|
|
| 23 |
|
| 24 |
# Set the working directory to the user's home directory
|
| 25 |
WORKDIR $HOME/app
|
| 26 |
|
|
|
|
|
|
|
| 27 |
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
|
| 28 |
COPY --chown=user . $HOME/app
|
| 29 |
|
|
|
|
| 19 |
GRADIO_NUM_PORTS=1 \
|
| 20 |
GRADIO_SERVER_NAME=0.0.0.0 \
|
| 21 |
GRADIO_THEME=huggingface \
|
| 22 |
+
SYSTEM=spaces \
|
| 23 |
+
HF_HOME=/data/.huggingface
|
| 24 |
|
| 25 |
# Set the working directory to the user's home directory
|
| 26 |
WORKDIR $HOME/app
|
| 27 |
|
| 28 |
+
CMD ["transformers-cli", "download", "/tiiuae/falcon-7b/" "--trust-remote-code"]
|
| 29 |
+
|
| 30 |
# Copy the current directory contents into the container at $HOME/app setting the owner to the user
|
| 31 |
COPY --chown=user . $HOME/app
|
| 32 |
|