bartduis commited on
Commit
6d1305b
·
verified ·
1 Parent(s): 32a956d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -32,11 +32,11 @@ dino_model = torch.hub.load('facebookresearch/dinov2', "dinov2_vitl14_reg")
32
  dino_model.eval()
33
  dino_model.to(device)
34
 
35
- print("Loading RaySt3R model")
36
- rayst3r_checkpoint = hf_hub_download("bartduis/rayst3r", "rayst3r.pth")
37
- rayst3r_model = EvalWrapper(rayst3r_checkpoint,device='cpu')
38
- rayst3r_model = rayst3r_model.to(device)
39
- print("Loaded all models")
40
 
41
  # print("Loading MoGe model")
42
  # # Load the model from huggingface hub (or load from local).
 
32
  dino_model.eval()
33
  dino_model.to(device)
34
 
35
+ # print("Loading RaySt3R model")
36
+ # rayst3r_checkpoint = hf_hub_download("bartduis/rayst3r", "rayst3r.pth")
37
+ # rayst3r_model = EvalWrapper(rayst3r_checkpoint,device='cpu')
38
+ # rayst3r_model = rayst3r_model.to(device)
39
+ # print("Loaded all models")
40
 
41
  # print("Loading MoGe model")
42
  # # Load the model from huggingface hub (or load from local).