hank1996 commited on
Commit
ef7395c
·
1 Parent(s): a65c271

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -69,6 +69,7 @@ def detect(img,model):
69
  # Initialize
70
  set_logging()
71
  device = select_device(opt.device)
 
72
  half = device.type != 'cpu' # half precision only supported on CUDA
73
 
74
  # Load model
 
69
  # Initialize
70
  set_logging()
71
  device = select_device(opt.device)
72
+ print(device)
73
  half = device.type != 'cpu' # half precision only supported on CUDA
74
 
75
  # Load model