Config ini

config.ini

@@ -0,0 +1,15 @@
+import herramientas
+llave = herramientas.obtenAccesoHF()
+
+[hugging_face]
+previo = "Una fotografía de un plato blanco con "
+seconds_available = 1500
+espacio = "black-forest-labs/FLUX.1-schnell"
+inferencia = "black-forest-labs/FLUX.1-dev"
+proveedor =  "hf-inference"
+
+; config = configparser.ConfigParser()
+; config.read("config.ini")
+
+; # Configure JWT settings
+; SECRET_KEY = config["misc-keys"]["secret-jwt"]

funciones.py

@@ -6,8 +6,6 @@ import globales
 
 def genera_platillo_gpu(platillo):
 
-    client = gradio_client.Client(globales.espacio, hf_token=globales.llave)
-
     prompt = globales.previo + platillo
 
     print("Eso es el prompt final:", prompt)
@@ -15,12 +13,13 @@ def genera_platillo_gpu(platillo):
     kwargs = {
             "prompt": prompt,
             "api_name": "/infer"
-        }
-    
+        }    
 
     try:
+
+        client = gradio_client.Client(globales.espacio, hf_token=globales.llave)
         result = client.predict(**kwargs
-        # prompt=prompt,
+                                # prompt=prompt,
         # negative_prompt="",
         # seed=42,
         # randomize_seed=True,
@@ -28,14 +27,14 @@ def genera_platillo_gpu(platillo):
         # height=1024,
         # guidance_scale=3.5,
         # num_inference_steps=28,
-        # api_name="/infer"
+        # api_name="/infer"       
         )
         
         return result[0]
 
-    except Exception as e:
+    except Exception as e:     
         print("Excepción es: ", e)
-        return {"Error":"500"}
+        return "default.png"
 
 
 def genera_platillo_inference(platillo):
@@ -51,11 +50,12 @@ def genera_platillo_inference(platillo):
         image = client.text_to_image(
         prompt,
         model=globales.inferencia,
-        #seed=42, #default varía pero el default es que siempre sea la misma.
+        #seed=42, 
         #guidance_scale=7.5,
         #num_inference_steps=50,
         #width=1024, #El default es 1024 x 1024 y quizá 1024*768, el max es 1536. 
         #height=1024 #El límite de replicate es 1024.
+        
         )
 
         img_io = io.BytesIO()
@@ -66,4 +66,7 @@ def genera_platillo_inference(platillo):
         
     except Exception as e:
         print("Excepción es: ", e)
-        return {"Error":"500"}
+        with open("default.png", "rb") as default_image_file:
+            img_io_default = io.BytesIO(default_image_file.read())
+            img_io_default.seek(0)
+            return img_io_default

globales.py

@@ -3,7 +3,7 @@ import herramientas
 previo = "Una fotografía de un plato blanco con "
 llave = herramientas.obtenAccesoHF()
 
-seconds_available = 0
+seconds_available = 1500
 
 espacio = "black-forest-labs/FLUX.1-schnell"
 #espacio = "black-forest-labs/FLUX.1-dev"