vicoman commited on
Commit
c64aa7a
1 Parent(s): f7ba278

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ import gradio as gr
6
  learn = load_learner('export.pkl')
7
 
8
  # Definimos las etiquetas de nuestro modelo
9
- labels = learn.dls.vocab
10
-
11
 
12
  # Definimos una función que se encarga de llevar a cabo las predicciones
13
  def predict(img):
 
6
  learn = load_learner('export.pkl')
7
 
8
  # Definimos las etiquetas de nuestro modelo
9
+ labels = ["Edad 1","Edad 2","Edad 3","Edad 4","Edad 5","Edad 6","Edad 7"]
 
10
 
11
  # Definimos una función que se encarga de llevar a cabo las predicciones
12
  def predict(img):