Spaces:

PablitoGil14
/

Entregable3

Runtime error

PablitoGil14 commited on May 18, 2024

Commit

c24a618

verified ·

1 Parent(s): efe2bbb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,7 @@ repo_id = "PablitoGil14/ModelEmotions"
 learner = from_pretrained_fastai(repo_id)
 # Definimos las etiquetas de nuestro modelo
-labels = ['0','1','2','3']
 example1 = "I cant believe this happened."
 example2 = "Thats fantastic news!"

 learner = from_pretrained_fastai(repo_id)
 # Definimos las etiquetas de nuestro modelo
+labels = learner.dls.vocab
 example1 = "I cant believe this happened."
 example2 = "Thats fantastic news!"