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

app.py

@@ -98,7 +98,7 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
         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 
+        return generated_image_path, used_seed, used_prompt 
 
     except Exception as e:
         gr.Warning("Error: " + str(e))
@@ -177,7 +177,7 @@ def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_infer
     
     client = InferenceClient()
     refined_image = client.image_to_image(
-        image[0], 
+        image, 
         prompt=prompt, 
         negative_prompt=negative_prompt, 
         num_inference_steps=num_inference_steps, 
@@ -186,7 +186,7 @@ def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_infer
         model=model,
         strength=strength
     )
-    return [image[0], refined_image]
+    return [image, refined_image]
 
 
 css="""
@@ -339,7 +339,7 @@ with gr.Blocks(css=css) as demo:
 
     with gr.Row():
         with gr.Column():
-            generated_image = ImageSlider(label="Image / Refined Image", type="filepath")
+            generated_image = Image(label="Image / Refined Image")
             enhace_button = gr.Button("Enhance Image")
 
         with gr.Column():