Update app.py

app.py

@@ -21,7 +21,7 @@ concat = lambda x: np.concatenate(x, axis=0)
 
 # Embeddings
 gdown.cached_download(
-    url="https://drive.google.com/uc?id=116CiA_cXciGSl72tbAUDoN-f1B9Frp89",
+    url="https://huggingface.co/datasets/XAI/CHM-Corr-Data/resolve/main/embeddings.pickle",
     path="./embeddings.pickle",
     quiet=False,
     md5="002b2a7f5c80d910b9cc740c2265f058",
@@ -35,7 +35,7 @@ gdown.download(id="1SDtq6ap7LPPpYfLbAxaMGGmj0EAV_m_e")
 
 # CUB training set
 gdown.cached_download(
-    url="https://drive.google.com/uc?id=1iR19j7532xqPefWYT-BdtcaKnsEokIqo",
+    url="https://huggingface.co/datasets/XAI/CHM-Corr-Data/resolve/main/CUB_train.zip",
     path="./CUB_train.zip",
     quiet=False,
     md5="1bd99e73b2fea8e4c2ebcb0e7722f1b1",
@@ -50,7 +50,7 @@ torchvision.datasets.utils.extract_archive(
 
 # CHM Weights
 gdown.cached_download(
-    url="https://drive.google.com/uc?id=1yM1zA0Ews2I8d9-BGc6Q0hIAl7LzYqr0",
+    url="https://huggingface.co/datasets/XAI/CHM-Corr-Data/resolve/main/pas_psi.pt",
     path="pas_psi.pt",
     quiet=False,
     md5="6b7b4d7bad7f89600fac340d6aa7708b",