Update raw.py

raw.py

@@ -23,7 +23,7 @@ pipe = FluxControlNetPipeline.from_pretrained(
 pipe.to("cuda")
 
 @spaces.GPU
-def generate_image(prompt, scale, steps, control_image):
+def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_scale, guidance_scale):
     # Load control image
     control_image = load_image(control_image)
     w, h = control_image.size
@@ -33,9 +33,9 @@ def generate_image(prompt, scale, steps, control_image):
     image = pipe(
         prompt=prompt,
         control_image=control_image,
-        controlnet_conditioning_scale=0.6,
+        controlnet_conditioning_scale=controlnet_conditioning_scale,
         num_inference_steps=steps,
-        guidance_scale=3.5,
+        guidance_scale=guidance_scale,
         height=control_image.size[1],
         width=control_image.size[0]
     ).images[0]
@@ -49,6 +49,8 @@ iface = gr.Interface(
         gr.Slider(1, 3, value=1, label="Scale"),
         gr.Slider(6, 30, value=8, label="Steps"),
         gr.Image(type="pil", label="Control Image"),
+        gr.Slider(0, 1, value=0.6, label="ControlNet Scale"),
+        gr.Slider(1, 20, value=3.5, label="Guidance Scale"),
     ],
     outputs=[
         gr.Image(type="pil", label="Generated Image", format="png"),