Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,13 +12,22 @@ labels = learner.dls.vocab
|
|
12 |
def predict(text):
|
13 |
output = learner.predict(text)
|
14 |
print("Output of learner.predict:", output)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
15 |
if pred == '0':
|
16 |
return "Positive"
|
17 |
-
elif pred == 1:
|
18 |
return "Neutral"
|
19 |
else:
|
20 |
return "Negative"
|
21 |
|
|
|
22 |
examp = ["Todo el mundo con la segunda temporada de Narcos y yo no, PORQUE NO TENGO NETFLIX @user pasaros una suscripcion de 5 dias",
|
23 |
"Me gusta estar sola y me molesta la gente que habla mucho. Si tan solo nos junt谩ramos a escuchar m煤sica y de rato en rato hablar de ello",
|
24 |
"La voz sepsi que debe tener Que dicha Gaby! Ya el lunes regresar谩 con toda la pata"]
|
|
|
12 |
def predict(text):
|
13 |
output = learner.predict(text)
|
14 |
print("Output of learner.predict:", output)
|
15 |
+
if len(output) == 3:
|
16 |
+
pred, _, probs = output
|
17 |
+
elif len(output) == 2:
|
18 |
+
pred, probs = output
|
19 |
+
else:
|
20 |
+
raise ValueError("Unexpected output format from learner.predict")
|
21 |
+
|
22 |
+
# Convertir la predicci贸n a etiqueta
|
23 |
if pred == '0':
|
24 |
return "Positive"
|
25 |
+
elif pred == '1':
|
26 |
return "Neutral"
|
27 |
else:
|
28 |
return "Negative"
|
29 |
|
30 |
+
|
31 |
examp = ["Todo el mundo con la segunda temporada de Narcos y yo no, PORQUE NO TENGO NETFLIX @user pasaros una suscripcion de 5 dias",
|
32 |
"Me gusta estar sola y me molesta la gente que habla mucho. Si tan solo nos junt谩ramos a escuchar m煤sica y de rato en rato hablar de ello",
|
33 |
"La voz sepsi que debe tener Que dicha Gaby! Ya el lunes regresar谩 con toda la pata"]
|