Update app.py
Browse files
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,
|
| 11 |
-
max_new_tokens=40,
|
| 12 |
-
temperature=1.1,
|
| 13 |
-
top_p=0.9,
|
| 14 |
-
do_sample=True,
|
| 15 |
-
repetition_penalty=1.3,
|
| 16 |
-
num_return_sequences=1,
|
| 17 |
-
device=-1
|
| 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
|
| 25 |
-
Tu estilo es poesía en
|
| 26 |
-
|
| 27 |
-
|
|
|
|
| 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(
|
| 56 |
-
return {"input":
|
|
|
|
| 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}
|