Spaces:
Build error
Build error
Update app.py
Browse files
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(
|
16 |
#img = PILImage.create(img)
|
17 |
-
pred,pred_idx,probs = learner.predict(
|
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.
|