Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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')
|
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']
|