Update Dockerfile

Dockerfile

@@ -24,10 +24,10 @@ ENV HOME=/home/user \
  
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
-
-CMD ["transformers-cli", "download", "tiiuae/falcon-7b", "--trust-remote-code"]
-
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
+# CMD ["transformers-cli", "download", "tiiuae/falcon-7b", "--trust-remote-code"]
+RUN python -c 'from huggingface_hub import snapshot_download; snapshot_download("tiiuae/falcon-40b")'
+
 CMD ["python", "app.py"]