ronnief1 commited on
Commit
fa6e350
·
1 Parent(s): de534e0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ import torch
7
  import albumentations as albu
8
  from torch.utils.data import DataLoader
9
  from torch.utils.data import Dataset as BaseDataset
10
- from catalyst import SupervisedRunner
11
  import segmentation_models_pytorch as smp
12
  from io import StringIO
13
 
 
7
  import albumentations as albu
8
  from torch.utils.data import DataLoader
9
  from torch.utils.data import Dataset as BaseDataset
10
+ from catalyst.dl import SupervisedRunner
11
  import segmentation_models_pytorch as smp
12
  from io import StringIO
13