maviced commited on
Commit
ecdee40
·
verified ·
1 Parent(s): a324ce2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -88,7 +88,7 @@ class SegmentationAlbumentationsTransform(ItemTransform):
88
  aug = self.aug(image=np.array(img), mask=np.array(mask))
89
  return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
90
 
91
- repo_id = "maviced/Practica3-2"
92
  learn = from_pretrained_fastai(repo_id)
93
  model = learn.model
94
  model = model.cpu()
 
88
  aug = self.aug(image=np.array(img), mask=np.array(mask))
89
  return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
90
 
91
+ repo_id = "maviced/practica3"
92
  learn = from_pretrained_fastai(repo_id)
93
  model = learn.model
94
  model = model.cpu()