Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -81,14 +81,11 @@ def detect(img,model):
|
|
81 |
print(weights)
|
82 |
stride =32
|
83 |
model = torch.jit.load(weights)
|
|
|
84 |
|
85 |
model = model.to(device)
|
86 |
print(111111111)
|
87 |
-
|
88 |
-
|
89 |
-
if half:
|
90 |
-
model.half() # to FP16
|
91 |
-
|
92 |
|
93 |
# Set Dataloader
|
94 |
vid_path, vid_writer = None, None
|
|
|
81 |
print(weights)
|
82 |
stride =32
|
83 |
model = torch.jit.load(weights)
|
84 |
+
print(model)
|
85 |
|
86 |
model = model.to(device)
|
87 |
print(111111111)
|
88 |
+
|
|
|
|
|
|
|
|
|
89 |
|
90 |
# Set Dataloader
|
91 |
vid_path, vid_writer = None, None
|