Update app.py

app.py

@@ -10,7 +10,9 @@ import spaces
 
 DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
 print(f"Is CUDA available: {torch.cuda.is_available()}")
-print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+if torch.cuda.is_available():
+    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+    
 model_choice = {
     'SAM': None, 
     'HQ-SAM': None,