Moibe commited on
Commit
0983bee
1 Parent(s): 9de3c9a
Files changed (2) hide show
  1. funciones.py +10 -2
  2. globales.py +1 -0
funciones.py CHANGED
@@ -49,13 +49,21 @@ def genera_platillo_inference(platillo):
49
  modelo_actual = conexion_firebase.obtenDato('nowme', 'huggingface', 'modelo_actual')
50
  modelo = modelo_actual
51
 
52
- #print("Modelo:", modelo)
 
 
 
 
 
 
 
 
53
 
54
  prompt = globales.previo + platillo
55
  print("Platillo enviado:", platillo)
56
 
57
  client = InferenceClient(
58
- provider= globales.proveedor,
59
  api_key=globales.llave
60
  )
61
 
 
49
  modelo_actual = conexion_firebase.obtenDato('nowme', 'huggingface', 'modelo_actual')
50
  modelo = modelo_actual
51
 
52
+ print("Modelo:", modelo)
53
+
54
+ #Elegir proveedor de inferencia
55
+ creditos_restantes_inference = conexion_firebase.obtenDato('nowme', 'huggingface', 'hfInference')
56
+ if creditos_restantes_inference > 0:
57
+ provedor_seleccionado = globales.proveedor
58
+ else:
59
+ provedor_seleccionado = globales.proveedor_back
60
+
61
 
62
  prompt = globales.previo + platillo
63
  print("Platillo enviado:", platillo)
64
 
65
  client = InferenceClient(
66
+ provider= provedor_seleccionado,
67
  api_key=globales.llave
68
  )
69
 
globales.py CHANGED
@@ -9,5 +9,6 @@ espacio = "black-forest-labs/FLUX.1-schnell"
9
  #inferencia = "black-forest-labs/FLUX.1-dev"
10
  #inferencia_backup = "black-forest-labs/FLUX.1-schnell"
11
  proveedor = "hf-inference" #falai
 
12
  work_cost = 6 #Los segundos que cuesta la generaci贸n de una imagen, para control de la capa gratuita.
13
  quota = 1500
 
9
  #inferencia = "black-forest-labs/FLUX.1-dev"
10
  #inferencia_backup = "black-forest-labs/FLUX.1-schnell"
11
  proveedor = "hf-inference" #falai
12
+ proveedor_back = "falai"
13
  work_cost = 6 #Los segundos que cuesta la generaci贸n de una imagen, para control de la capa gratuita.
14
  quota = 1500