Update Dockerfile

Dockerfile

@@ -23,7 +23,8 @@ ENV HOME=/home/user \
 	SYSTEM=spaces
 
 # download falcon-40b
-RUN transformers-cli download "tiiuae/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