Update app.py

app.py

@@ -8,6 +8,8 @@ import torch
 from huggingface_hub import hf_hub_download
 
 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())}")
 model_choice = {
     'SAM': None, 
     'HQ-SAM': None,