Moibe commited on
Commit
ac6309b
·
1 Parent(s): bb5a2a3

Servicio Secundario para Procesamiento de Lotes

Browse files
Files changed (1) hide show
  1. app.py +26 -1
app.py CHANGED
@@ -51,4 +51,29 @@ async def genera_imagen(platillo: str = Form(...)):
51
  if type(resultado) is str:
52
  return resultado
53
  else:
54
- return StreamingResponse(content=resultado, media_type="image/png")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  if type(resultado) is str:
52
  return resultado
53
  else:
54
+ return StreamingResponse(content=resultado, media_type="image/png")
55
+
56
+ @app.post("/procesador-lotes/")
57
+ async def procesa_lote(platillo: str = Form(...)):
58
+
59
+ print("Servicio Secundario de Procesamiento de Lotes")
60
+ #Obtengo los segundos disponibles de procesamiento para saber si tengo GPU disponible de la capa gratuita o me voy a método cobrado.
61
+ seconds_available = herramientas.obtenSegundosDisponibles()
62
+ print(herramientas.imprimeTimeNow())
63
+
64
+ if seconds_available > globales.work_cost:
65
+ print("Usando GPU (capa gratuita)...")
66
+ resultado = funciones.genera_platillo_gpu(platillo)
67
+ if "Error" in resultado:
68
+ return resultado
69
+ else:
70
+ return FileResponse(resultado, media_type="image/png", filename="imagen.png")
71
+ else:
72
+
73
+ print("Usando Inference...")
74
+ resultado = funciones.genera_platillo_inference(platillo)
75
+ print("El resultado de inference es: ", resultado)
76
+ if type(resultado) is str:
77
+ return resultado
78
+ else:
79
+ return StreamingResponse(content=resultado, media_type="image/png")