Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -19,8 +19,8 @@ hf_hub_download("merve/siglip-faiss-wikiart", "wikiart_10k.csv", local_dir="./")
|
|
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("
|
23 |
-
model = SiglipModel.from_pretrained("
|
24 |
|
25 |
def read_image_from_url(url):
|
26 |
response = requests.get(url)
|
|
|
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)
|
24 |
|
25 |
def read_image_from_url(url):
|
26 |
response = requests.get(url)
|