Spaces:

lightly-ai
/

food101-image-retrieval

Running

SauravMaheshkar commited on Jun 27

Commit

358b6e7

•

1 Parent(s): e5a7396

fix: var name

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ transforms = T.Compose(
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-datset = load_dataset("ethz/food101")
 model = DINO(batch_size_per_device=32, num_classes=1000).to(device)
 model.load_state_dict(torch.load("./bin/model.ckpt", map_location=device)["state_dict"])

 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
+dataset = load_dataset("ethz/food101")
 model = DINO(batch_size_per_device=32, num_classes=1000).to(device)
 model.load_state_dict(torch.load("./bin/model.ckpt", map_location=device)["state_dict"])