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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -4
app.py CHANGED
@@ -29,9 +29,17 @@ 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
37
  app = FastAPI(title="Golemet API FastAPI")
@@ -44,5 +52,5 @@ def root():
44
  # Ruta reemplazando tu /golemet
45
  @app.get("/golemet")
46
  def procesar_cadena(cadena: str = Query("???", description="Texto a generar")):
47
- texto_generado = gpt(cadena)
48
  return {"input": cadena, "output": texto_generado}
 
29
  Respuesta:
30
  """.strip()
31
 
32
+ raw = generator(prompt)[0]["generated_text"]
33
+
34
+ # (Opcional) Recortar solo la parte tras "Respuesta:"
35
+ respuesta = raw.split("Respuesta:")[-1].strip()
36
+
37
+ # (Opcional) Ajuste leve para intentar aproximar 26 palabras
38
+ palabras = respuesta.split()
39
+ if len(palabras) > 26:
40
+ respuesta = " ".join(palabras[:26])
41
+
42
+ return respuesta
43
 
44
  # Crear app FastAPI
45
  app = FastAPI(title="Golemet API FastAPI")
 
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}