hank1996 commited on
Commit
1370b21
·
1 Parent(s): 94de35e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def detect(img,model):
85
  #stride = int(model.stride.max()) # model stride
86
  #imgsz = check_img_size(imgsz, s=stride) # check img_size
87
  if weights == 'yolopv2.pt':
88
-
89
  stride =32
90
  model = torch.jit.load(weights,map_location=device)
91
  model.eval()
 
85
  #stride = int(model.stride.max()) # model stride
86
  #imgsz = check_img_size(imgsz, s=stride) # check img_size
87
  if weights == 'yolopv2.pt':
88
+ print(weights)
89
  stride =32
90
  model = torch.jit.load(weights,map_location=device)
91
  model.eval()