golemet commited on
Commit
c2e1751
·
verified ·
1 Parent(s): 0a520be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -14
app.py CHANGED
@@ -7,24 +7,25 @@ model_name = "flax-community/gpt-2-spanish"
7
  # Inicializa Pipeline configurado
8
  generator = pipeline(
9
  "text-generation",
10
- model=model_name, # Puedes cambiar a otro modelo compatible
11
- max_new_tokens=40, # Suficiente para ~26 palabras sin desbordarse
12
- temperature=1.1, # Favorece creatividad y metáforas
13
- top_p=0.9, # Diversidad controlada
14
- do_sample=True, # Permite variación creativa
15
- repetition_penalty=1.3, # Reduce repeticiones
16
- num_return_sequences=1, # Solo una respuesta
17
- device=-1 # CPU, recomendado en HuggingFace Spaces
18
  )
19
 
20
  # Prompt base para moldear el estilo y tono
21
  def generar_respuesta_reflexiva(pregunta):
22
  prompt = f"""
23
  Eres una voz femenina que responde con aire conversacional y reflexivo.
24
- Hablas con metáforas, analogías y juegos de contrarios.
25
- Tu estilo es poesía en prosa mezclada con aforismo.
26
- Genera exactamente una respuesta breve, cercana a 26 palabras.
27
- Evita repeticiones. Termina siempre con una pregunta abierta.
 
28
  Pregunta: {pregunta}
29
  Respuesta:
30
  """.strip()
@@ -52,5 +53,5 @@ def root():
52
  # Ruta reemplazando tu /golemet
53
  @app.get("/golemet")
54
  def procesar_cadena(cadena: str = Query("???", description="Texto a generar")):
55
- texto_generado = generar_respuesta_reflexiva(cadena)
56
- return {"input": cadena, "output": texto_generado}
 
7
  # Inicializa Pipeline configurado
8
  generator = pipeline(
9
  "text-generation",
10
+ model=model_name, # modelo compatible
11
+ max_new_tokens=40, # Suficiente para ~26 palabras sin desbordarse
12
+ temperature=1.1, # Favorece creatividad y metáforas
13
+ top_p=0.9, # Diversidad controlada
14
+ do_sample=True, # Permite variación creativa
15
+ repetition_penalty=1.3, # Reduce repeticiones
16
+ num_return_sequences=1, # Solo una respuesta
17
+ device=-1 # CPU, recomendado en HuggingFace Spaces
18
  )
19
 
20
  # Prompt base para moldear el estilo y tono
21
  def generar_respuesta_reflexiva(pregunta):
22
  prompt = f"""
23
  Eres una voz femenina que responde con aire conversacional y reflexivo.
24
+ Hablas con metáforas, analogías y comprendes la ley de los contrarios,
25
+ estableciendo una síntesis dialéctica profunda. Tu estilo es poesía en
26
+ prosa sentenciosa, con un carácter de aforismo y de visión oracular.
27
+ Genera exactamente una respuesta breve, cercana a 26 palabras, que
28
+ termine en un adjetivo. Evita repeticiones.
29
  Pregunta: {pregunta}
30
  Respuesta:
31
  """.strip()
 
53
  # Ruta reemplazando tu /golemet
54
  @app.get("/golemet")
55
  def procesar_cadena(cadena: str = Query("???", description="Texto a generar")):
56
+ texto_generado = generar_respuesta_reflexiva(texto)
57
+ return {"input": texto, "output": texto_generado}