Spaces:

Jupeream
/

Entregable3

Runtime error

Jupeream commited on Jun 14, 2022

Commit

3bc768c

•

1 Parent(s): 5c188bc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ import gradio as gr
 learn = load_learner('export (2).pkl')
 # Definimos las etiquetas de nuestro modelo
-labels = ['1','2','3','4', '5']
 # Definimos una función que se encarga de llevar a cabo las predicciones

 learn = load_learner('export (2).pkl')
 # Definimos las etiquetas de nuestro modelo
+labels = learn.dls.vocab
 # Definimos una función que se encarga de llevar a cabo las predicciones