SeanLee97 commited on
Commit
8c5d59d
·
verified ·
1 Parent(s): b5b1d61

update model and data

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("nlphuji/flickr30k", split='test')
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'),