Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ embeds_sexual = modelo_vector.encode(frases_sexual)
|
|
54 |
def verificar_semantico(descripcion):
|
55 |
emb_desc = modelo_vector.encode(descripcion)
|
56 |
tipos_detectados = []
|
57 |
-
if max(util.cos_sim(emb_desc, embeds_fisica)[0]) > 0.
|
58 |
tipos_detectados.append("física")
|
59 |
if max(util.cos_sim(emb_desc, embeds_sexual)[0]) > 0.7:
|
60 |
tipos_detectados.append("sexual")
|
@@ -113,12 +113,12 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
|
|
113 |
"CONSIDERACIONES:\n"
|
114 |
"- Hechos relevantes y valoración del riesgo.\n"
|
115 |
"- Fundamentos jurídicos aplicables.\n"
|
116 |
-
"- Justificación breve de cada literal del Artículo 5
|
117 |
"RESUELVE:\n"
|
118 |
"PRIMERO: Admitir la solicitud.\n"
|
119 |
-
"SEGUNDO
|
120 |
-
"
|
121 |
-
"
|
122 |
)
|
123 |
}
|
124 |
]
|
@@ -136,7 +136,14 @@ def predecir_con_recomendacion(edad, genero, hijos, convivencia_agresor, consumo
|
|
136 |
|
137 |
# Decodificar
|
138 |
decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
139 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
140 |
|
141 |
return tipos_str, riesgo, tipo_medida, auto_redactado
|
142 |
|
|
|
54 |
def verificar_semantico(descripcion):
|
55 |
emb_desc = modelo_vector.encode(descripcion)
|
56 |
tipos_detectados = []
|
57 |
+
if max(util.cos_sim(emb_desc, embeds_fisica)[0]) > 0.8:
|
58 |
tipos_detectados.append("física")
|
59 |
if max(util.cos_sim(emb_desc, embeds_sexual)[0]) > 0.7:
|
60 |
tipos_detectados.append("sexual")
|
|
|
113 |
"CONSIDERACIONES:\n"
|
114 |
"- Hechos relevantes y valoración del riesgo.\n"
|
115 |
"- Fundamentos jurídicos aplicables.\n"
|
116 |
+
"- Justificación breve de cada literal aplicable del Artículo 5 de la Ley 575 de 2000, identificando por letra (a), (b), etc.\n\n"
|
117 |
"RESUELVE:\n"
|
118 |
"PRIMERO: Admitir la solicitud.\n"
|
119 |
+
"SEGUNDO, TERCERO, CUARTO, etc.: Ordenar cada medida de manera separada, en un párrafo por medida.\n"
|
120 |
+
"ÚLTIMO: Cúmplase y notifíquese.\n\n"
|
121 |
+
"Redacta en un estilo formal y claro, máximo 400 palabras."
|
122 |
)
|
123 |
}
|
124 |
]
|
|
|
136 |
|
137 |
# Decodificar
|
138 |
decoded = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
|
139 |
+
auto_completo = decoded[0].strip()
|
140 |
+
|
141 |
+
# Recortar desde "CONSIDERACIONES"
|
142 |
+
inicio = auto_completo.find("CONSIDERACIONES")
|
143 |
+
if inicio != -1:
|
144 |
+
auto_redactado = auto_completo[inicio:]
|
145 |
+
else:
|
146 |
+
auto_redactado = auto_completo
|
147 |
|
148 |
return tipos_str, riesgo, tipo_medida, auto_redactado
|
149 |
|