merve HF staff commited on
Commit
74ba09f
1 Parent(s): ccb786c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ hf_hub_download("merve/siglip-faiss-wikiart", "siglip_10k_latest.index", local_d
16
  hf_hub_download("merve/siglip-faiss-wikiart", "wikiart_10k_latest.csv", local_dir="./")
17
 
18
  # read index, dataset and load siglip model and processor
19
- index = faiss.read_index("./siglip_10k.index")
20
- df = pd.read_csv("./wikiart_10k.csv")
21
  device = torch.device('cuda' if torch.cuda.is_available() else "cpu")
22
  processor = AutoProcessor.from_pretrained("google/siglip-base-patch16-224")
23
  model = SiglipModel.from_pretrained("google/siglip-base-patch16-224").to(device)
 
16
  hf_hub_download("merve/siglip-faiss-wikiart", "wikiart_10k_latest.csv", local_dir="./")
17
 
18
  # read index, dataset and load siglip model and processor
19
+ index = faiss.read_index("./siglip_10k_latest.index")
20
+ df = pd.read_csv("./wikiart_10k_latest.csv")
21
  device = torch.device('cuda' if torch.cuda.is_available() else "cpu")
22
  processor = AutoProcessor.from_pretrained("google/siglip-base-patch16-224")
23
  model = SiglipModel.from_pretrained("google/siglip-base-patch16-224").to(device)