Spaces:
Running
Running
update model and data
Browse files
app.py
CHANGED
@@ -104,9 +104,9 @@ if __name__ == "__main__":
|
|
104 |
from datasets import load_dataset
|
105 |
from huggingface_hub import snapshot_download
|
106 |
|
107 |
-
local_dir = snapshot_download(repo_id='WhereIsAI/angle-flickr-index')
|
108 |
|
109 |
-
ds = load_dataset("
|
110 |
search = FlickrAngleSearch.from_preindexed(
|
111 |
os.path.join(local_dir, 'index.faiss'),
|
112 |
os.path.join(local_dir, 'captions.pt'),
|
|
|
104 |
from datasets import load_dataset
|
105 |
from huggingface_hub import snapshot_download
|
106 |
|
107 |
+
local_dir = snapshot_download(repo_id='WhereIsAI/angle-flickr-index-2')
|
108 |
|
109 |
+
ds = load_dataset("WhereIsAI/flickr30k-v2", split='train')
|
110 |
search = FlickrAngleSearch.from_preindexed(
|
111 |
os.path.join(local_dir, 'index.faiss'),
|
112 |
os.path.join(local_dir, 'captions.pt'),
|