allopeap commited on
Commit
bd17c33
verified
1 Parent(s): 31230ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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"]