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):
|