huntrezz commited on
Commit
7b83683
·
verified ·
1 Parent(s): 7bc8ed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ model = torch.quantization.quantize_dynamic(
29
 
30
  model = model.half().to(device)
31
 
32
- processor = DPTImageProcessor.from_pretrained("Intel/dpt-swinv2-tiny-128")
33
 
34
  color_map = cv2.applyColorMap(np.arange(256, dtype=np.uint8), cv2.COLORMAP_INFERNO)
35
  color_map = torch.from_numpy(color_map).to(device)
 
29
 
30
  model = model.half().to(device)
31
 
32
+ processor = DPTImageProcessor.from_pretrained("Intel/dpt-swinv2-tiny-256")
33
 
34
  color_map = cv2.applyColorMap(np.arange(256, dtype=np.uint8), cv2.COLORMAP_INFERNO)
35
  color_map = torch.from_numpy(color_map).to(device)