JUGGHM commited on
Commit
885c69a
1 Parent(s): 8afac0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ def depth_normal(img):
72
  pred_color = gray_to_colormap(pred_depth)
73
 
74
  pred_normal = pred_normal.squeeze()
75
- if pred_normal.size(0) == 3
76
  pred_normal == pred_normal.permute(1,2,0)
77
  pred_color_normal = vis_surface_normal(pred_normal)
78
 
 
72
  pred_color = gray_to_colormap(pred_depth)
73
 
74
  pred_normal = pred_normal.squeeze()
75
+ if pred_normal.size(0) == 3:
76
  pred_normal == pred_normal.permute(1,2,0)
77
  pred_color_normal = vis_surface_normal(pred_normal)
78