We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f8e4e35 commit eb18751Copy full SHA for eb18751
1 file changed
apps/dash-dino/app.py
@@ -68,6 +68,7 @@ def apply_threshold(attentions, w_featmap, h_featmap, threshold, patch_size=16):
68
default_url = "https://dl.fbaipublicfiles.com/dino/img.png"
69
70
# Load model
71
+torch.hub.set_dir("./")
72
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
73
print("Running on", device)
74
model = torch.hub.load("facebookresearch/dino:main", "dino_deits16").to(device)
0 commit comments