Lisandro commited on
Commit
f70d5f1
·
1 Parent(s): 75378b7

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

Browse files

Los cambios en el código mejoran el proceso de refinamiento de imágenes en el archivo `app.py`. La función `refine_image` ahora utiliza el cliente de inferencia de Hugging Face para manejar la transformación de imagen a imagen. Esto mejora la eficiencia y confiabilidad del proceso de refinamiento.

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,7 +6,7 @@ from gradio_imageslider import ImageSlider
6
 
7
 
8
  def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
9
- image = client.image_to_image(
10
  image,
11
  prompt=prompt,
12
  negative_prompt=negative_prompt,
@@ -16,7 +16,7 @@ def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_s
16
  model="stabilityai/stable-diffusion-xl-refiner-1.0",
17
  strength=strength
18
  )
19
- return image
20
 
21
  with gr.Blocks() as demo:
22
  image = gr.Image(type="filepath")
@@ -52,7 +52,7 @@ with gr.Blocks() as demo:
52
  value=-1
53
  )
54
  refine_btn = gr.Button("Refine")
55
- output = gr.Image()
56
 
57
  refine_btn.click(
58
  refine_image,
 
6
 
7
 
8
  def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
9
+ refined_image = client.image_to_image(
10
  image,
11
  prompt=prompt,
12
  negative_prompt=negative_prompt,
 
16
  model="stabilityai/stable-diffusion-xl-refiner-1.0",
17
  strength=strength
18
  )
19
+ return [image, refined_image]
20
 
21
  with gr.Blocks() as demo:
22
  image = gr.Image(type="filepath")
 
52
  value=-1
53
  )
54
  refine_btn = gr.Button("Refine")
55
+ output = ImageSlider(label="Before / After")
56
 
57
  refine_btn.click(
58
  refine_image,