Spaces:

fffiloni
/

CLIP-Interrogator-2

Running on T4

bartman081523 commited on Mar 23, 2023

Commit

55af7b4

1 Parent(s): 4d736e9

update urls and wget --continue

update urls and wget --continue
#23 #24 #25 #26

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,15 +18,15 @@ setup()
 # download cache files
 print("Download preprocessed cache files...")
 CACHE_URLS = [
-    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_artists.pkl',
-    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_flavors.pkl',
-    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_mediums.pkl',
-    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_movements.pkl',
-    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_trendings.pkl',
 ]
 os.makedirs('cache', exist_ok=True)
 for url in CACHE_URLS:
-    print(subprocess.run(['wget', url, '-P', 'cache'], stdout=subprocess.PIPE).stdout.decode('utf-8'))
 import sys
 sys.path.append('src/blip')

 # download cache files
 print("Download preprocessed cache files...")
 CACHE_URLS = [
+    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_artists.safetensors',
+    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_flavors.safetensors',
+    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_mediums.safetensors',
+    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_movements.safetensors',
+    'https://huggingface.co/pharma/ci-preprocess/resolve/main/ViT-H-14_laion2b_s32b_b79k_trendings.safetensors',
 ]
 os.makedirs('cache', exist_ok=True)
 for url in CACHE_URLS:
+    print(subprocess.run(['wget', url, '-P', '-c', 'cache'], stdout=subprocess.PIPE).stdout.decode('utf-8'))
 import sys
 sys.path.append('src/blip')