feat: Update image refinement process in app.py

app.py

@@ -1,22 +1,34 @@
 import os
 import gradio as gr
-
 from huggingface_hub import InferenceClient
 client = InferenceClient()
-
 from gradio_imageslider import ImageSlider
 
-stable_diffusion_xl_refiner_10 = "https://api-inference.huggingface.co/models/stabilityai/stable-diffusion-xl-refiner-1.0"
-
-def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed):
 
-    image = client.image_to_image(image, prompt=prompt, negative_prompt=negative_prompt, num_inference_steps=num_inference_steps, guidance_scale=guidance_scale, seed=seed, model="stabilityai/stable-diffusion-xl-refiner-1.0")
+def refine_image(image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
+    image = client.image_to_image(
+        image, 
+        prompt=prompt, 
+        negative_prompt=negative_prompt, 
+        num_inference_steps=num_inference_steps, 
+        guidance_scale=guidance_scale, 
+        seed=seed, 
+        model="stabilityai/stable-diffusion-xl-refiner-1.0",
+        strength=strength
+    )
     return image
 
 with gr.Blocks() as demo:
     image = gr.Image(type="filepath")
     prompt = gr.Textbox(lines=3, label="Prompt")
     negative_prompt = gr.Textbox(lines=3, label="Negative Prompt")
+    strength = gr.Slider(
+        label="Strength",
+        minimum=0,
+        maximum=3,
+        step=0.01,
+        value=1
+    )
     num_inference_steps = gr.Slider(
         label="Inference steps",
         minimum=3,
@@ -42,7 +54,11 @@ with gr.Blocks() as demo:
     refine_btn = gr.Button("Refine")
     output = gr.Image()
 
-    refine_btn.click(refine_image, inputs=[image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed], outputs=output)
+    refine_btn.click(
+        refine_image, 
+        inputs=[image, prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength], 
+        outputs=output
+    )
 
 demo.launch()