thejarll commited on
Commit
5d9d076
·
verified ·
1 Parent(s): fb241bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -56,6 +56,7 @@ llm = pipeline(
56
  )
57
 
58
  @GPU # DECORADOR para que ZeroGPU ejecute esta función en CPU/GPU remota
 
59
  def responder(pregunta):
60
  if not pregunta:
61
  return "Por favor ingresa una pregunta."
@@ -68,7 +69,7 @@ def responder(pregunta):
68
  palabras_expandidas = expandir_con_sinonimos(palabras_clave, diccionario_sinonimos)
69
 
70
  filtrados = [c for c in result_chunks if any(p in c.lower() for p in palabras_expandidas)]
71
- contexto_final = "\n\n".join(filtrados[:3]) if filtrados else "\n\n".join(result_chunks[:3])
72
 
73
  prompt = f"""
74
  Eres un abogado colombiano especializado en el Código de Tránsito, Código de Policía y Código Penal.
@@ -95,7 +96,7 @@ def responder(pregunta):
95
  resultado = llm(
96
  prompt,
97
  max_new_tokens=500,
98
- temperature=0.4,
99
  top_p=0.9,
100
  repetition_penalty=1.2
101
  )[0]["generated_text"]
 
56
  )
57
 
58
  @GPU # DECORADOR para que ZeroGPU ejecute esta función en CPU/GPU remota
59
+
60
  def responder(pregunta):
61
  if not pregunta:
62
  return "Por favor ingresa una pregunta."
 
69
  palabras_expandidas = expandir_con_sinonimos(palabras_clave, diccionario_sinonimos)
70
 
71
  filtrados = [c for c in result_chunks if any(p in c.lower() for p in palabras_expandidas)]
72
+ contexto_final = "\n\n".join(filtrados[:10]) if filtrados else "\n\n".join(result_chunks[:10])
73
 
74
  prompt = f"""
75
  Eres un abogado colombiano especializado en el Código de Tránsito, Código de Policía y Código Penal.
 
96
  resultado = llm(
97
  prompt,
98
  max_new_tokens=500,
99
+ temperature=0.7,
100
  top_p=0.9,
101
  repetition_penalty=1.2
102
  )[0]["generated_text"]