vop020506 commited on
Commit
6cd6769
1 Parent(s): cce545e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -8,8 +8,9 @@ from fastai.vision.all import *
8
  repo_id = "vop020506/entregable3"
9
 
10
  learner = from_pretrained_fastai(repo_id)
11
- labels = learner.dls.vocab
12
-
 
13
  # Definimos una función que se encarga de llevar a cabo las predicciones
14
  def predict(img):
15
  #img = PILImage.create(img)
 
8
  repo_id = "vop020506/entregable3"
9
 
10
  learner = from_pretrained_fastai(repo_id)
11
+ labels = ['admiration','amusement', 'anger', 'annoyance', 'approval', 'caring','confusion','curiosity','desire', 'disappointment',
12
+ 'disapproval', 'disgust', 'embarrassment', 'excitement', 'fear', 'gratitude', 'grief', 'joy', 'love', 'nervousness', 'optimism', 'pride',
13
+ 'realization','relief', 'remorse', 'sadness', 'surprise', 'neutral']
14
  # Definimos una función que se encarga de llevar a cabo las predicciones
15
  def predict(img):
16
  #img = PILImage.create(img)