panda1835 commited on
Commit
48f78b9
·
1 Parent(s): f00a620

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import json
7
  from ultralytics import YOLO
8
 
9
  print(f"Is CUDA available: {torch.cuda.is_available()}")
10
- print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
 
12
  # Load the model
13
  # detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)
 
7
  from ultralytics import YOLO
8
 
9
  print(f"Is CUDA available: {torch.cuda.is_available()}")
10
+ # print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
11
 
12
  # Load the model
13
  # detect_model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, _verbose=False)