Update app.py

app.py

@@ -32,11 +32,8 @@ examples = [
     ['assets/example_videos/sheep_cut1.mp4', -1, -1, 1280, True, True, True, 0.3],
 ]
 
-if not torch.cuda.is_available():
-    sys.exit("CUDA is not available. Exiting...")
-
-DEVICE = 'cuda'
-print(f"Using device: {DEVICE}")
+# Use GPU if available; otherwise, use CPU.
+DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 # Model configuration for different encoder variants.
 model_configs = {