Spaces:
Sleeping
Sleeping
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'),
|