Update app.py
Browse files
app.py
CHANGED
@@ -67,7 +67,7 @@ import torchvision.transforms as transforms
|
|
67 |
|
68 |
# learner = from_pretrained_fastai(repo_id)
|
69 |
|
70 |
-
learn = unet_learner(None,resnet18,cbs=
|
71 |
learn.load('model')
|
72 |
|
73 |
def transform_image(image):
|
|
|
67 |
|
68 |
# learner = from_pretrained_fastai(repo_id)
|
69 |
|
70 |
+
learn = unet_learner(None,resnet18,cbs=None,metrics=[DiceMulti()]).to_fp16()
|
71 |
learn.load('model')
|
72 |
|
73 |
def transform_image(image):
|