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):
|