Update app.py

app.py

@@ -24,8 +24,8 @@ def load_model(config_path, model_path, encoder, device):
 # Inference function
 def depth_estimation(image, model_path, encoder='vits'):
     try:
-        # device = 'cuda' if torch.cuda.is_available() else 'cpu'
-        device = 'cpu'
+        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        # device = 'cpu'
         config_path = 'configs/config_v2_vits14.json'
         
         # Ensure model path exists or download if needed