RaulCtm commited on
Commit
0c0fc2e
·
verified ·
1 Parent(s): 66a0795

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -35,5 +35,5 @@ def predecir_toxicidad(peticion: PeticionComentario):
35
  "texto_analizado": peticion.texto,
36
  "etiqueta_modelo": resultado['label'],
37
  "score_confianza": round(float(resultado['score']), 4),
38
- "es_toxico": True if resultado['label'] == "Toxic" else False
39
- }
 
35
  "texto_analizado": peticion.texto,
36
  "etiqueta_modelo": resultado['label'],
37
  "score_confianza": round(float(resultado['score']), 4),
38
+ "es_toxico": True if resultado['label'] in ["Toxic", "NEGATIVE"] else False
39
+ }