PablitoGil14 commited on
Commit
cf8d99f
·
verified ·
1 Parent(s): 441c086

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -72,7 +72,7 @@ class SegmentationAlbumentationsTransform(ItemTransform):
72
  return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
73
 
74
  #Cargamos el modelo
75
- repo_id = "luisvarona/Practica3"
76
  learn = from_pretrained_fastai(repo_id)
77
  model = learn.model
78
  model = model.cpu()
 
72
  return PILImage.create(aug["image"]), PILMask.create(aug["mask"])
73
 
74
  #Cargamos el modelo
75
+ repo_id = "PablitoGil14/AP-Practica3"
76
  learn = from_pretrained_fastai(repo_id)
77
  model = learn.model
78
  model = model.cpu()