feat: Agregar función de carga de imágenes en app.py

Se ha agregado una nueva función llamada `load_img` en el archivo app.py. Esta función permite cargar imágenes desde una ruta de archivo y devuelve la imagen cargada en formato de cadena. Esta función se utiliza para cargar la imagen generada antes de devolverla como resultado de la inferencia.

app.py

@@ -6,6 +6,7 @@ from gradio_client import Client, handle_file
 from gradio_imageslider import ImageSlider
 from PIL import Image
 from huggingface_hub import InferenceClient
+from loadimg import load_img
 
 with open('loras.json', 'r') as f:
     loras = json.load(f)
@@ -95,6 +96,7 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
         generated_image_path = result[0]  # Esto puede necesitar ser ajustado basado en la estructura real de result
         used_seed = result[1]             # Esto puede necesitar ser ajustado basado en la estructura real de result
         used_prompt = prompt              # El prompt usado es simplemente el prompt procesado
+        generated_image_path = load_img(generated_image_path, output_type="str")
 
         return [generated_image_path, generated_image_path], used_seed, used_prompt 
 

requirements.txt

@@ -4,4 +4,5 @@
 # torch
 # transformers
 # xformers
-gradio-imageslider
+gradio-imageslider
+loadimg