Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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("./
|
20 |
-
df = pd.read_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)
|