Update app.py

app.py

@@ -80,7 +80,7 @@ def depth_normal(img, model_selection="vit-small"):
 
         pred_normal = output['normal_out_list'][0][:, :3, :, :] 
         H, W = pred_normal.shape[2:]
-        pred_normal[:, :, pad[0]:H-pad[1], pad[2]:W-pad[3]]
+        pred_normal = pred_normal[:, :, pad[0]:H-pad[1], pad[2]:W-pad[3]]
 
     pred_depth = pred_depth.squeeze().cpu().numpy()
     pred_depth[pred_depth<0] = 0