lichorosario commited on
Commit
7e809df
1 Parent(s): 0f1fc3f

feat: Refactorizar función de refinamiento de imágenes en app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -171,7 +171,7 @@ def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidanc
171
 
172
  def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
173
  if (not apply_refiner):
174
- return [image, image]
175
 
176
  client = InferenceClient()
177
  refined_image = client.image_to_image(
@@ -184,7 +184,7 @@ def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_infer
184
  model=model,
185
  strength=strength
186
  )
187
- return [image, refined_image]
188
 
189
 
190
  css="""
 
171
 
172
  def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
173
  if (not apply_refiner):
174
+ return image
175
 
176
  client = InferenceClient()
177
  refined_image = client.image_to_image(
 
184
  model=model,
185
  strength=strength
186
  )
187
+ return refined_image
188
 
189
 
190
  css="""