maviced commited on
Commit
65bed86
·
verified ·
1 Parent(s): 464aed5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -88,8 +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"
92
- repo_id ="lauragordo/model3"
93
  learn = from_pretrained_fastai(repo_id)
94
  model = learn.model
95
  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-2"
 
92
  learn = from_pretrained_fastai(repo_id)
93
  model = learn.model
94
  model = model.cpu()