LPX55 commited on
Commit
87f19f6
·
verified ·
1 Parent(s): f845c63

Update raw.py

Browse files
Files changed (1) hide show
  1. raw.py +6 -4
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
- control_image = gr.Image(type="pil", label="Control Image")
68
  controlnet_conditioning_scale = gr.Slider(0, 1, value=0.6, label="ControlNet Scale")
69
- generate_button = gr.Button("Generate Image", variant="primary")
70
 
71
  with gr.Row():
72
- generated_image = gr.Image(type="pil", label="Generated Image", format="png")
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,