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)
|