Update app.py

app.py

@@ -6,6 +6,9 @@ import torch
 import json
 from ultralytics import YOLO
 
+print(f"Is CUDA available: {torch.cuda.is_available()}")
+print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
+
 # Load the model
 # detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
 detect_model = YOLO('yolov8m_2023-10-23_best.pt')