Update app.py

app.py

@@ -1,15 +1,16 @@
 import os
-import subprocess
+from huggingface_hub import Repository
 
 # Retrieve the token from the environment variables
 token = os.environ.get("token")
 
-# Clone the repository using the token
-repo_url = f"https://robocan:{token}@huggingface.co/robocan/GeoG_City"
-destination_dir = os.path.expanduser("~/SVD")  # Use a directory in the home directory
-
-# Run the git clone command using subprocess
-subprocess.run(["git", "clone", repo_url, destination_dir], check=True)
+repo = Repository(
+    local_dir="SVD",
+    repo_type="model",
+    clone_from="robocan/GeoG_City",
+    token=token
+)
+repo.git_pull()
 
 import torch
 from torch.utils.data import Dataset, DataLoader