Update run.sh

run.sh

@@ -6,26 +6,9 @@ MODEL_URL_="${MODEL_URL:-https://huggingface.co/TheBloke/Mixtral-8x7B-v0.1-GGUF/
 PORT="${PORT:-7860}"
 CONTEXT="${CONTEXT:-32768}"
 
-# Check if the file exists
-if [ ! -e "/data/models/command-r-plus-Q6_K-00001-of-00002.gguf" ]; then
-    echo "Downloading model."
-    mkdir -p /data/models
-    wget -x -q -O /data/models/command-r-plus-Q6_K-00001-of-00002.gguf "https://huggingface.co/pmysl/c4ai-command-r-plus-GGUF/resolve/main/command-r-plus-Q6_K-00001-of-00002.gguf"
-    echo "Part 1 downloaded. Restart space to download part 2"
-    exit
-else
-    echo "File already exists."
-fi
-# Check if the file exists
-if [ ! -e "/data/models/command-r-plus-Q6_K-00002-of-00002.gguf" ]; then
-    echo "Downloading model."
-    mkdir -p /data/models
-    wget -x -q -O /data/models/command-r-plus-Q6_K-00002-of-00002.gguf "https://huggingface.co/pmysl/c4ai-command-r-plus-GGUF/resolve/main/command-r-plus-Q6_K-00002-of-00002.gguf"
-    echo "Part 2 downloaded. Restart space to run app"
-    exit
-else
-    echo "File already exists."
-fi
+huggingface-cli download "pmysl/c4ai-command-r-plus-GGUF" command-r-plus-Q6_K-00001-of-00002.gguf ---local-dir /data/models
+huggingface-cli download "pmysl/c4ai-command-r-plus-GGUF" command-r-plus-Q6_K-00002-of-00002.gguf ---local-dir /data/models
+
 #--n-gpu-layers $N_GPU_LAYERS
 #-c $CONTEXT
 cd /app