wildanzm commited on
Commit
a2ddd41
·
verified ·
1 Parent(s): 01d406e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,13 +7,13 @@ import warnings
7
 
8
  import torch
9
  from ultralytics.nn.tasks import ClassificationModel
10
- from ultralytics.nn.modules importconv, block
11
  from torch.nn import Sequential, Conv2d, Module
12
 
13
  torch.serialization.add_safe_globals([
14
  ClassificationModel,
15
  Sequential,
16
- Conv,
17
  Conv2d,
18
  block.C2f,
19
  block.Bottleneck,
 
7
 
8
  import torch
9
  from ultralytics.nn.tasks import ClassificationModel
10
+ from ultralytics.nn.modules import conv, block
11
  from torch.nn import Sequential, Conv2d, Module
12
 
13
  torch.serialization.add_safe_globals([
14
  ClassificationModel,
15
  Sequential,
16
+ conv.Conv,
17
  Conv2d,
18
  block.C2f,
19
  block.Bottleneck,