Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,8 +6,6 @@ import torch
|
|
6 |
import torchvision.models as models
|
7 |
from torch import nn
|
8 |
|
9 |
-
from model import *
|
10 |
-
|
11 |
from albumentations import (
|
12 |
HorizontalFlip, VerticalFlip, IAAPerspective, ShiftScaleRotate, CLAHE, RandomRotate90,
|
13 |
Transpose, ShiftScaleRotate, Blur, OpticalDistortion, GridDistortion, HueSaturationValue,
|
|
|
6 |
import torchvision.models as models
|
7 |
from torch import nn
|
8 |
|
|
|
|
|
9 |
from albumentations import (
|
10 |
HorizontalFlip, VerticalFlip, IAAPerspective, ShiftScaleRotate, CLAHE, RandomRotate90,
|
11 |
Transpose, ShiftScaleRotate, Blur, OpticalDistortion, GridDistortion, HueSaturationValue,
|