Update raw.py

raw.py

@@ -20,17 +20,17 @@ pipe = FluxControlNetPipeline.from_pretrained(
 pipe.to("cuda")
 
 @spaces.GPU
-def generate_image(prompt, control_image):
+def generate_image(prompt, scale, steps, control_image):
     # Load control image
     control_image = load_image(control_image)
     w, h = control_image.size
     # Upscale x1
-    control_image = control_image.resize((w * 1, h * 1))
+    control_image = control_image.resize((w * scale, h * scale))
     image = pipe(
         prompt=prompt,
         control_image=control_image,
         controlnet_conditioning_scale=0.6,
-        num_inference_steps=8,
+        num_inference_steps=steps,
         guidance_scale=3.5,
         height=control_image.size[1],
         width=control_image.size[0]
@@ -42,6 +42,8 @@ iface = gr.Interface(
     fn=generate_image,
     inputs=[
         gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+        gr.Slider(1, 3, value=1, label="Scale"),
+        gr.Slider(6, 30, value=8, label="Steps"),
         gr.Image(type="pil", label="Control Image"),
     ],
     outputs=[