Update Dockerfile

Dockerfile

@@ -22,12 +22,12 @@ ENV HOME=/home/user \
 	GRADIO_THEME=huggingface \
 	SYSTEM=spaces
 
-# download falcon-40b
-RUN python -c "from transformers import AutoModel; AutoModel.from_pretrained('tiiuae/falcon-40b', trust_remote_code=True)"
- 
  
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
+# download falcon-40b
+RUN python -c "from transformers import AutoModel; AutoModel.from_pretrained('tiiuae/falcon-40b', trust_remote_code=True)"
+ 
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app