Update app.py

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import matplotlib.cm as cm
 import matplotlib  # New import for the updated colormap API
 import subprocess
+import sys
 
 from video_depth_anything.video_depth import VideoDepthAnything
 from utils.dc_utils import read_video_frames, save_video
@@ -31,8 +32,11 @@ examples = [
     ['assets/example_videos/sheep_cut1.mp4', -1, -1, 1280, True, True, True, 0.3],
 ]
 
-# Use GPU if available; otherwise, use CPU.
-DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
+if not torch.cuda.is_available():
+    sys.exit("CUDA is not available. Exiting...")
+
+DEVICE = 'cuda'
+print(f"Using device: {DEVICE}")
 
 # Model configuration for different encoder variants.
 model_configs = {