Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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']
|
| 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 |
+
}
|