dokcer tweaks

Dockerfile

@@ -13,7 +13,7 @@ RUN corepack enable \
     && corepack prepare yarn@latest --activate
 
 # Install huggingface_hub and huggingface_hub[cli]
-RUN pip install -U "huggingface_hub[cli]"
+RUN pip install -U "huggingface_hub"
 
 # Define a working directory
 WORKDIR /app

build.sh

@@ -9,11 +9,8 @@ if [ -z "$HF_TOKEN" ] || [ -z "$HF_SPACE_NAME" ] || [ -z "$BUILD_COMMAND" ] || [
     exit 1
 fi
 
-# Set the Hugging Face token
-export HF_HUB_ACCESS_TOKEN=$HF_TOKEN
-
 # Download the space
-huggingface-cli download spaces/$HF_SPACE_NAME -d space
+huggingface-cli download $HF_SPACE_NAME --repo-type space --local-dir space --token $HF_TOKEN
 
 # Change directory to the space's directory
 cd space
@@ -39,6 +36,6 @@ eval $BUILD_COMMAND
 echo "Build completed successfully, uploading the build output..."
 
 # Upload the build output to the Hugging Face space
-huggingface-cli upload spaces/$HF_SPACE_NAME $OUTPUT_PATH / --revision refs/convert/build --token $HF_TOKEN
+huggingface-cli upload $HF_SPACE_NAME $OUTPUT_PATH / --repo-type space --revision refs/convert/build --token $HF_TOKEN
 
 echo "Build and upload completed successfully."