Update app.py

app.py

@@ -44,6 +44,7 @@ def generar_embedding(texto):
         mask = inputs["attention_mask"].unsqueeze(-1).expand(last_hidden.size()).float()
         summed = torch.sum(last_hidden * mask, 1)
         counted = torch.clamp(mask.sum(1), min=1e-9)
+
         mean_pooled = summed / counted
         return mean_pooled.numpy()
 
@@ -71,18 +72,15 @@ def responder(pregunta):
     contexto_final = "\n\n".join(filtrados[:15]) if filtrados else "\n\n".join(result_chunks[:15])
 
     prompt = f"""
-Eres un abogado colombiano especializado en derecho de tránsito, policía y penal.
-Tu única fuente de información es el siguiente CONTEXTO LEGAL.  
-No tienes permitido inventar artículos, leyes o consecuencias que no estén citadas o explícitas en el contexto.
-
-INSTRUCCIONES:
-- Fundamenta tu respuesta exclusivamente en el CONTEXTO LEGAL.
-- Si mencionas consecuencias, citas o artículos, deben estar explícitamente en el texto.
-- No incluyas enlaces web, imágenes, disclaimers institucionales ni logos.
-- No agregues conclusiones si no están fundamentadas.
-- Usa un lenguaje claro, profesional, breve y entendible para una persona sin conocimientos jurídicos.
-- Si no encuentras nada útil en el contexto, responde:
-"No encontré información suficiente en los documentos para responder esta pregunta."
+Eres un asistente legal colombiano especializado en normas de tránsito, código de policía y código penal. Tu única fuente de información es el texto legal a continuación.
+
+Instrucciones:
+- Lee cuidadosamente el contexto legal proporcionado.
+- Responde la pregunta basándote únicamente en ese contexto. No inventes información ni agregues artículos que no estén en el texto.
+- No cites números de artículos a menos que estén explícitamente escritos en el contexto.
+- No uses frases como “esperamos que esto sea útil”, “hasta luego” ni conclusiones formales.
+- Si no encuentras información relevante, responde de forma clara: **"No encontré información suficiente en los documentos para responder esta pregunta."**
+- Escribe de forma clara, concisa y orientada a una persona sin formación legal.
 
 CONTEXTO LEGAL:
 {contexto_final}
@@ -93,12 +91,13 @@ PREGUNTA:
 RESPUESTA:
 """
 
+
     resultado = llm(
         prompt,
         max_new_tokens=500,
         temperature=0.4,
         top_p=0.9,
-        repetition_penalty=1.3
+        repetition_penalty=1.2
     )[0]["generated_text"]
 
     if "RESPUESTA:" in resultado: