golemet commited on
Commit
c03822b
·
verified ·
1 Parent(s): 1896e62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -8
app.py CHANGED
@@ -4,20 +4,33 @@ from transformers import pipeline
4
  # Modelo español
5
  model_name = "flax-community/gpt-2-spanish"
6
 
7
- # Inicializa el pipeline
8
  generator = pipeline(
9
  "text-generation",
10
- model=model_name,
 
 
 
 
 
 
11
  device=-1 # CPU, recomendado en HuggingFace Spaces
12
  )
13
 
 
 
 
 
 
 
 
 
 
 
 
 
14
  def gpt(texto: str):
15
- result = generator(
16
- texto,
17
- do_sample=True,
18
- pad_token_id=50256,
19
- max_length=30
20
- )[0]["generated_text"]
21
  return result
22
 
23
  # Crear app FastAPI
 
4
  # Modelo español
5
  model_name = "flax-community/gpt-2-spanish"
6
 
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()
31
+
32
  def gpt(texto: str):
33
+ result = generator(prompt)[0]["generated_text"]
 
 
 
 
 
34
  return result
35
 
36
  # Crear app FastAPI