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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -80,8 +80,7 @@ def detect(img,model):
80
  print(weights)
81
  stride =32
82
  model = torch.jit.load(weights)
83
- device = select_device(opt.device)
84
- half = device.type != 'cpu' # half precision only supported on CUDA
85
  model = model.to(device)
86
  print(111111111)
87
 
 
80
  print(weights)
81
  stride =32
82
  model = torch.jit.load(weights)
83
+
 
84
  model = model.to(device)
85
  print(111111111)
86