Update app.py

app.py

@@ -5,7 +5,7 @@ from transformers import DPTForDepthEstimation, DPTImageProcessor
 import gradio as gr
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = DPTForDepthEstimation.from_pretrained("./", local_files_only=True, torch_dtype=torch.float16).to(device)
+model = DPTForDepthEstimation.from_pretrained("Intel/dpt-swinv2-tiny-256", torch_dtype=torch.float16).to(device)
 model = torch.jit.script(model)
 processor = DPTImageProcessor.from_pretrained("Intel/dpt-swinv2-tiny-256")