penway commited on
Commit
3bc7e2d
·
1 Parent(s): 65156b8

add token for DINO

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -8,6 +8,9 @@ import matplotlib.pyplot as plt
8
  from weights import W, b
9
 
10
  # Load DINOv3
 
 
 
11
  MODEL_ID = "facebook/dinov3-vitl16-pretrain-lvd1689m"
12
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
13
  model = AutoModel.from_pretrained(MODEL_ID)
 
8
  from weights import W, b
9
 
10
  # Load DINOv3
11
+ token = os.environ.get("DINO_TOKEN")
12
+ if token:
13
+ login(token)
14
  MODEL_ID = "facebook/dinov3-vitl16-pretrain-lvd1689m"
15
  processor = AutoImageProcessor.from_pretrained(MODEL_ID)
16
  model = AutoModel.from_pretrained(MODEL_ID)