bartduis commited on
Commit
4ab1a0e
·
verified ·
1 Parent(s): 6d1305b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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):