merve HF staff commited on
Commit
fcc4e47
1 Parent(s): b53eff8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("nielsr/siglip-base-patch16-224")
23
- model = SiglipModel.from_pretrained("nielsr/siglip-base-patch16-224").to(device)
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)