Update app.py
Browse files
app.py
CHANGED
|
@@ -28,9 +28,9 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
| 28 |
outdir = "/tmp/rayst3r"
|
| 29 |
|
| 30 |
moge_model = MoGeModel.from_pretrained("Ruicheng/moge-vitl").to(device)
|
| 31 |
-
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")
|
|
@@ -204,8 +204,8 @@ def process_image(input_img):
|
|
| 204 |
input_glb = input_to_glb(outdir,input_img,depth_dict,mask,rotated=rotated)
|
| 205 |
print('Input done')
|
| 206 |
print('calling Ray')
|
| 207 |
-
inference_glb = rayst3r_to_glb(input_img,depth_dict,mask,rotated=rotated)
|
| 208 |
-
print(input_glb)
|
| 209 |
return input_img, input_img
|
| 210 |
|
| 211 |
# def process_image(input_img):
|
|
|
|
| 28 |
outdir = "/tmp/rayst3r"
|
| 29 |
|
| 30 |
moge_model = MoGeModel.from_pretrained("Ruicheng/moge-vitl").to(device)
|
| 31 |
+
# 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")
|
|
|
|
| 204 |
input_glb = input_to_glb(outdir,input_img,depth_dict,mask,rotated=rotated)
|
| 205 |
print('Input done')
|
| 206 |
print('calling Ray')
|
| 207 |
+
# inference_glb = rayst3r_to_glb(input_img,depth_dict,mask,rotated=rotated)
|
| 208 |
+
# print(input_glb)
|
| 209 |
return input_img, input_img
|
| 210 |
|
| 211 |
# def process_image(input_img):
|