Update app.py
Browse files
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).
|