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,
|