Update app.py

app.py

@@ -7,7 +7,7 @@ import torch.nn.utils.prune as prune
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model = DPTForDepthEstimation.from_pretrained("Intel/dpt-swinv2-tiny-128", torch_dtype=torch.float16)
+model = DPTForDepthEstimation.from_pretrained("Intel/dpt-swinv2-tiny-256", torch_dtype=torch.float16)
 model.eval()
 
 # Apply global unstructured pruning