AdrianRevi commited on
Commit
e5a5a0f
·
verified ·
1 Parent(s): d2d9897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ dls = ImageDataLoaders.from_name_func(
18
  # 2. Crea el learner y carga el modelo desde .pth
19
  # ---------------------------------------------------------
20
  learn = cnn_learner(dls, resnet18, metrics=accuracy)
21
- learn.load('resnet18_blindness') # asegúrate de que el archivo .pth esté en la raíz o models/
22
 
23
  # Define tus clases manualmente si no están en dls.vocab
24
  labels = ['No Blindness', 'Blindness']
 
18
  # 2. Crea el learner y carga el modelo desde .pth
19
  # ---------------------------------------------------------
20
  learn = cnn_learner(dls, resnet18, metrics=accuracy)
21
+ learn.load('resnet18_blindness')
22
 
23
  # Define tus clases manualmente si no están en dls.vocab
24
  labels = ['No Blindness', 'Blindness']