Moibe commited on
Commit
d90f087
·
1 Parent(s): e9bc1c2

Se agregó work_margin para req espacio HF

Browse files
Files changed (5) hide show
  1. app.py +2 -2
  2. funciones.py +1 -1
  3. globales.py +1 -0
  4. herramientas.py +2 -2
  5. stats.py +1 -1
app.py CHANGED
@@ -32,13 +32,13 @@ async def echo_image(image: UploadFile = File(...)):
32
  @app.post("/genera-imagen/")
33
  async def genera_imagen(request: Request, platillo: str = Form(...)):
34
 
35
- print("Client host:", request.client.host)
36
  herramientas.despliegaInfoCliente(request)
37
 
38
  seconds_available = herramientas.obtenSegundosDisponibles()
39
  print(herramientas.imprimeTimeNow())
40
 
41
- if seconds_available > globales.work_cost:
42
  print("Usando ZeroGPU 🪭.")
43
  resultado = funciones.genera_platillo_gpu(platillo)
44
  if "Error" in resultado:
 
32
  @app.post("/genera-imagen/")
33
  async def genera_imagen(request: Request, platillo: str = Form(...)):
34
 
35
+ #print("Client host:", request.client.host)
36
  herramientas.despliegaInfoCliente(request)
37
 
38
  seconds_available = herramientas.obtenSegundosDisponibles()
39
  print(herramientas.imprimeTimeNow())
40
 
41
+ if seconds_available > globales.work_margin: #Funcionará si es mayor a 75s por el margen que HF requiere, la resta si usa el work_cost.
42
  print("Usando ZeroGPU 🪭.")
43
  resultado = funciones.genera_platillo_gpu(platillo)
44
  if "Error" in resultado:
funciones.py CHANGED
@@ -87,7 +87,7 @@ def genera_platillo_inference(platillo):
87
  # provedor_seleccionado = globales.proveedor_back
88
 
89
  prompt = globales.previo + platillo
90
- print("Platillo enviado:", platillo)
91
 
92
  client = InferenceClient(
93
  provider= provedor_seleccionado,
 
87
  # provedor_seleccionado = globales.proveedor_back
88
 
89
  prompt = globales.previo + platillo
90
+ #print("Platillo enviado:", platillo)
91
 
92
  client = InferenceClient(
93
  provider= provedor_seleccionado,
globales.py CHANGED
@@ -10,5 +10,6 @@ inferencia_backup = "black-forest-labs/FLUX.1-schnell"
10
  proveedor = "hf-inference" #falai
11
  proveedor_back = "fal-ai"
12
  work_cost = 6 #Los segundos que cuesta la generación de una imagen, para control de la capa gratuita.
 
13
  inference_cost = 1
14
  quota = 1500
 
10
  proveedor = "hf-inference" #falai
11
  proveedor_back = "fal-ai"
12
  work_cost = 6 #Los segundos que cuesta la generación de una imagen, para control de la capa gratuita.
13
+ work_margin = 75
14
  inference_cost = 1
15
  quota = 1500
herramientas.py CHANGED
@@ -47,7 +47,7 @@ def restaSegundosGPU(cuantos_segundos):
47
 
48
  # Restar los segundos
49
  nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
50
- print("Procesado, segundos disponibles ahora: ", nuevos_segundos_disponibles)
51
  conexion_firebase.editaDato('nowme', servidor, 'segundos', nuevos_segundos_disponibles)
52
 
53
  def restaSegundosInference(cuantos_segundos):
@@ -123,7 +123,7 @@ def siEsDiaSiguienteRenueva():
123
  renuevaSegundosDisponibles()
124
  renuevaTimestampActual()
125
  else:
126
- print("Aún no hay renovación de capa de procesamiento.")
127
  pass
128
  else:
129
  print("Aún no hay renovación de capa de procesamiento.")
 
47
 
48
  # Restar los segundos
49
  nuevos_segundos_disponibles = segundos_disponibles - cuantos_segundos
50
+ #print("Procesado, segundos disponibles ahora: ", nuevos_segundos_disponibles)
51
  conexion_firebase.editaDato('nowme', servidor, 'segundos', nuevos_segundos_disponibles)
52
 
53
  def restaSegundosInference(cuantos_segundos):
 
123
  renuevaSegundosDisponibles()
124
  renuevaTimestampActual()
125
  else:
126
+ #print("Aún no hay renovación de capa de procesamiento.")
127
  pass
128
  else:
129
  print("Aún no hay renovación de capa de procesamiento.")
stats.py CHANGED
@@ -46,7 +46,7 @@ def sacar_estadistica_horas(coleccion_nombre):
46
  horas_creacion.append(hora)
47
 
48
  if not horas_creacion:
49
- print("❌ No se encontraron timestamps de creación para analizar.")
50
  return {}
51
 
52
  # 3. Contar la frecuencia de cada hora
 
46
  horas_creacion.append(hora)
47
 
48
  if not horas_creacion:
49
+ #print("❌ No se encontraron timestamps de creación para analizar.")
50
  return {}
51
 
52
  # 3. Contar la frecuencia de cada hora