Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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()
|