paascorb commited on
Commit
4b036ae
·
1 Parent(s): 48f9f45

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=callbacks,metrics=[DiceMulti()]).to_fp16()
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):