Update app.py

app.py

@@ -10,3 +10,4 @@ import yaml
 from huggingface_hub import hf_hub_download
 from ultralytics import YOLO
 
+device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')