thirarbi commited on
Commit
dd2fda6
·
verified ·
1 Parent(s): a892138

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,7 +20,8 @@ import subprocess
20
  from ultralytics.nn.tasks import DetectionModel
21
  from torch.nn import Sequential
22
  from ultralytics.nn.modules.conv import Conv
23
- torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv])
 
24
 
25
  from io import BytesIO
26
  import glob
 
20
  from ultralytics.nn.tasks import DetectionModel
21
  from torch.nn import Sequential
22
  from ultralytics.nn.modules.conv import Conv
23
+ mport the class: from torch.nn import Conv2d
24
+ torch.serialization.add_safe_globals([DetectionModel, Sequential, Conv, Conv2d])
25
 
26
  from io import BytesIO
27
  import glob