paascorb commited on
Commit
48f9f45
·
1 Parent(s): 907c6f8

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(trainDLS,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=callbacks,metrics=[DiceMulti()]).to_fp16()
71
  learn.load('model')
72
 
73
  def transform_image(image):