Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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-
|
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)
|