fix(ci): Use dvc pull by stage name and add dvc.lock

.github/workflows/main.yaml

@@ -25,7 +25,7 @@ jobs:
       - name: Install DVC and project requirements
         run: |
           pip install "dvc[gdrive]"
-          pip install -r requirements.txt # requirements.txt should also have huggingface_hub
+          pip install -r requirements.txt
 
       - name: Configure DVC
         env:
@@ -34,15 +34,14 @@ jobs:
           mkdir -p .dvc/tmp
           echo "$GDRIVE_CREDENTIALS_DATA" > .dvc/tmp/gdrive-user-credentials.json
 
+      # --- THIS IS THE FIX ---
       - name: Pull DVC Artifacts
-        run: dvc pull artifacts/model_training/model.dvc -f
+        run: dvc pull -s model_training -f
 
-      # --- THIS IS THE FIX ---
       - name: Log in to Hugging Face Hub
         env:
           HF_TOKEN: ${{ secrets.HF_TOKEN }}
         run: |
-          # huggingface_hub is already installed via requirements.txt, but this is a safe fallback
           pip install huggingface_hub
           huggingface-cli login --token $HF_TOKEN
 
@@ -54,7 +53,13 @@ jobs:
           git config --global user.email "action@github.com"
           git config --global user.name "GitHub Action"
           git clone https://huggingface.co/spaces/${HF_USERNAME}/${HF_SPACE_NAME} hf_space_repo
-          # Use rsync for a more robust copy
+          
+          # We need to copy the dvc.yaml and dvc.lock files for the pull to work in the previous step
+          # This is incorrect. The pull happens before the clone. Let's fix the logic.
+          # The checkout action already brings dvc.yaml and dvc.lock
+
+          # The directory structure will be correct after the dvc pull.
+          # We just need to copy the final app and the pulled model.
           rsync -av --delete app/ app.py requirements.txt README.md artifacts/model_training/model/ hf_space_repo/
           cd hf_space_repo
           git add .