Update app.py
Browse files
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/
|
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()
|