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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,9 +12,9 @@ labels = ['admiration','amusement', 'anger', 'annoyance', 'approval', 'caring','
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)
17
- pred,pred_idx,probs = learner.predict(img)
18
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
19
 
20
  # Creamos la interfaz y la lanzamos.
 
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(text):
16
  #img = PILImage.create(img)
17
+ pred,pred_idx,probs = learner.predict(text)
18
  return {labels[i]: float(probs[i]) for i in range(len(labels))}
19
 
20
  # Creamos la interfaz y la lanzamos.