oxkitsune commited on
Commit
ab53a86
1 Parent(s): 4d29a77

fix double numpy array

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ def predict(frame):
45
  image = image.to(device)
46
  prediction = model.infer(image)
47
  depth = prediction["depth"].squeeze().detach().cpu().numpy()
48
- return depth.cpu().numpy(), prediction["focallength_px"].item()
49
 
50
 
51
  @rr.thread_local_stream("rerun_example_ml_depth_pro")
 
45
  image = image.to(device)
46
  prediction = model.infer(image)
47
  depth = prediction["depth"].squeeze().detach().cpu().numpy()
48
+ return depth, prediction["focallength_px"].item()
49
 
50
 
51
  @rr.thread_local_stream("rerun_example_ml_depth_pro")