Update app.py

app.py

@@ -73,7 +73,7 @@ def depth_normal(img):
 
     pred_normal = pred_normal.squeeze()
     if pred_normal.size(0) == 3:
-        pred_normal == pred_normal.permute(1,2,0)
+        pred_normal = pred_normal.permute(1,2,0)
     pred_color_normal = vis_surface_normal(pred_normal)
     
     ##formatted = (output * 255 / np.max(output)).astype('uint8')