Update app.py

app.py

@@ -166,7 +166,7 @@ def input_to_glb(outdir,img,depth_dict,mask,rotated=False):
 def depth_moge(input_img):
     
     input_img_torch = torch.tensor(input_img / 255, dtype=torch.float32, device=device).permute(2, 0, 1)
-    output = moge_model.infer(input_img_torch)['depth'].cpu()
+    output = moge_model.infer(input_img_torch)
     return output 
 
 @GPU(duration = 180)