huntrezz commited on
Commit
37d103e
·
verified ·
1 Parent(s): 0f0d2de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -36,8 +36,6 @@ processor = DPTImageProcessor.from_pretrained("Intel/dpt-swinv2-tiny-256")
36
  color_map = cv2.applyColorMap(np.arange(256, dtype=np.uint8), cv2.COLORMAP_INFERNO)
37
  color_map = torch.from_numpy(color_map).to(device)
38
 
39
- visualizer = DepthVisualizer()
40
-
41
  def preprocess_image(image):
42
  image = cv2.resize(image, (128, 128))
43
  image = torch.from_numpy(image).permute(2, 0, 1).unsqueeze(0).float().to(device)
 
36
  color_map = cv2.applyColorMap(np.arange(256, dtype=np.uint8), cv2.COLORMAP_INFERNO)
37
  color_map = torch.from_numpy(color_map).to(device)
38
 
 
 
39
  def preprocess_image(image):
40
  image = cv2.resize(image, (128, 128))
41
  image = torch.from_numpy(image).permute(2, 0, 1).unsqueeze(0).float().to(device)