Spaces:
Running
on
Zero
Running
on
Zero
Update raw.py
Browse files
raw.py
CHANGED
@@ -39,7 +39,7 @@ def generate_image(prompt, scale, steps, control_image, controlnet_conditioning_
|
|
39 |
control_image = load_image(control_image)
|
40 |
w, h = control_image.size
|
41 |
# Upscale x1
|
42 |
-
control_image = control_image.resize((int(w * scale), int(h * scale)))
|
43 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
44 |
image = pipe(
|
45 |
prompt=prompt,
|
@@ -59,17 +59,19 @@ with gr.Blocks(title="FLUX ControlNet Image Generation") as iface:
|
|
59 |
|
60 |
with gr.Row():
|
61 |
with gr.Column(scale=1):
|
|
|
62 |
prompt = gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt")
|
63 |
scale = gr.Slider(1, 3, value=1, label="Scale")
|
64 |
steps = gr.Slider(6, 30, value=8, label="Steps")
|
65 |
guidance_scale = gr.Slider(1, 20, value=3.5, label="Guidance Scale")
|
|
|
66 |
with gr.Column(scale=1):
|
67 |
-
|
68 |
controlnet_conditioning_scale = gr.Slider(0, 1, value=0.6, label="ControlNet Scale")
|
69 |
-
|
70 |
|
71 |
with gr.Row():
|
72 |
-
|
73 |
|
74 |
generate_button.click(
|
75 |
fn=generate_image,
|
|
|
39 |
control_image = load_image(control_image)
|
40 |
w, h = control_image.size
|
41 |
# Upscale x1
|
42 |
+
control_image = control_image.resize((int(w * scale), int(h * scale)), resample=2) # Resample.BILINEAR
|
43 |
print("Size to: " + str(control_image.size[0]) + ", " + str(control_image.size[1]))
|
44 |
image = pipe(
|
45 |
prompt=prompt,
|
|
|
59 |
|
60 |
with gr.Row():
|
61 |
with gr.Column(scale=1):
|
62 |
+
control_image = gr.Image(type="pil", label="Control Image")
|
63 |
prompt = gr.Textbox(lines=2, placeholder="Enter your prompt here...", label="Prompt")
|
64 |
scale = gr.Slider(1, 3, value=1, label="Scale")
|
65 |
steps = gr.Slider(6, 30, value=8, label="Steps")
|
66 |
guidance_scale = gr.Slider(1, 20, value=3.5, label="Guidance Scale")
|
67 |
+
generate_button = gr.Button("Generate Image", variant="primary")
|
68 |
with gr.Column(scale=1):
|
69 |
+
generated_image = gr.Image(type="pil", label="Generated Image", format="png")
|
70 |
controlnet_conditioning_scale = gr.Slider(0, 1, value=0.6, label="ControlNet Scale")
|
71 |
+
|
72 |
|
73 |
with gr.Row():
|
74 |
+
gr.Markdown("**Tips:** Lorum ipsum")
|
75 |
|
76 |
generate_button.click(
|
77 |
fn=generate_image,
|