Spaces:
Sleeping
Sleeping
Add falai
Browse files- funciones.py +10 -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 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
prompt = globales.previo + platillo
|
| 55 |
print("Platillo enviado:", platillo)
|
| 56 |
|
| 57 |
client = InferenceClient(
|
| 58 |
-
provider=
|
| 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
|