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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
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