km1lo commited on
Commit
150bf43
·
verified ·
1 Parent(s): 29e28a9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
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.7:
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 aplicable.\n\n"
117
  "RESUELVE:\n"
118
  "PRIMERO: Admitir la solicitud.\n"
119
- "SEGUNDO: Ordenar las siguientes medidas (una por línea).\n"
120
- "TERCERO: Cúmplase y notifíquese.\n\n"
121
- "Limita tu respuesta a máximo 600 palabras, en estilo formal."
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
- auto_redactado = decoded[0].strip()
 
 
 
 
 
 
 
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