feat: Mejorar proceso de refinamiento de imágenes en app.py

app.py

@@ -5,7 +5,7 @@ client = InferenceClient()
 from gradio_imageslider import ImageSlider
 
 
-def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
+def refine_image(image, model ,prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
     refined_image = client.image_to_image(
         image, 
         prompt=prompt, 
@@ -13,7 +13,7 @@ def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_s
         num_inference_steps=num_inference_steps, 
         guidance_scale=guidance_scale, 
         seed=seed, 
-        model="stabilityai/stable-diffusion-xl-refiner-1.0",
+        model=model,
         strength=strength
     )
     return [image, refined_image]
@@ -23,6 +23,7 @@ with gr.Blocks() as demo:
         with gr.Column():
             image = gr.Image(type="filepath")
             with gr.Accordion("Advanced Options", open=False):
+                model = gr.Textbox(label="Model", default="stabilityai/stable-diffusion-xl-refiner-1.0")
                 prompt = gr.Textbox(lines=3, label="Prompt")
                 negative_prompt = gr.Textbox(lines=3, label="Negative Prompt")
                 strength = gr.Slider(
@@ -60,7 +61,7 @@ with gr.Blocks() as demo:
 
     refine_btn.click(
         refine_image, 
-        inputs=[image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength], 
+        inputs=[image, model, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength], 
         outputs=output
     )