Update Modules/Generate_Image.py
Browse files
Modules/Generate_Image.py
CHANGED
|
@@ -97,7 +97,7 @@ def build_interface() -> gr.Interface:
|
|
| 97 |
return gr.Interface(
|
| 98 |
fn=Generate_Image,
|
| 99 |
inputs=[
|
| 100 |
-
gr.Textbox(label="Prompt", placeholder="Enter a prompt", lines=2),
|
| 101 |
gr.Textbox(
|
| 102 |
label="Model",
|
| 103 |
value="black-forest-labs/FLUX.1-Krea-dev",
|
|
@@ -113,16 +113,18 @@ def build_interface() -> gr.Interface:
|
|
| 113 |
"mutated, ugly, disgusting, blurry, amputation, misspellings, typos"
|
| 114 |
),
|
| 115 |
lines=2,
|
|
|
|
| 116 |
),
|
| 117 |
-
gr.Slider(minimum=1, maximum=100, value=35, step=1, label="Steps"),
|
| 118 |
-
gr.Slider(minimum=1.0, maximum=20.0, value=7.0, step=0.1, label="CFG Scale"),
|
| 119 |
-
gr.Slider(minimum=-1, maximum=1_000_000_000, value=-1, step=1, label="Seed (-1 = random)"),
|
| 120 |
-
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Width"),
|
| 121 |
-
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Height"),
|
| 122 |
gr.Radio(
|
| 123 |
label="Sampler",
|
| 124 |
value="DPM++ 2M Karras",
|
| 125 |
choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"],
|
|
|
|
| 126 |
),
|
| 127 |
],
|
| 128 |
outputs=gr.Image(label="Generated Image"),
|
|
|
|
| 97 |
return gr.Interface(
|
| 98 |
fn=Generate_Image,
|
| 99 |
inputs=[
|
| 100 |
+
gr.Textbox(label="Prompt", placeholder="Enter a prompt", lines=2, info="Text description of the image to generate"),
|
| 101 |
gr.Textbox(
|
| 102 |
label="Model",
|
| 103 |
value="black-forest-labs/FLUX.1-Krea-dev",
|
|
|
|
| 113 |
"mutated, ugly, disgusting, blurry, amputation, misspellings, typos"
|
| 114 |
),
|
| 115 |
lines=2,
|
| 116 |
+
info="What should NOT appear in the image",
|
| 117 |
),
|
| 118 |
+
gr.Slider(minimum=1, maximum=100, value=35, step=1, label="Steps", info="Number of denoising steps (1–100)"),
|
| 119 |
+
gr.Slider(minimum=1.0, maximum=20.0, value=7.0, step=0.1, label="CFG Scale", info="Classifier-free guidance scale (1–20)"),
|
| 120 |
+
gr.Slider(minimum=-1, maximum=1_000_000_000, value=-1, step=1, label="Seed (-1 = random)", info="Random seed for reproducibility"),
|
| 121 |
+
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Width", info="Output width in pixels"),
|
| 122 |
+
gr.Slider(minimum=64, maximum=1216, value=1024, step=32, label="Height", info="Output height in pixels"),
|
| 123 |
gr.Radio(
|
| 124 |
label="Sampler",
|
| 125 |
value="DPM++ 2M Karras",
|
| 126 |
choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "Euler", "Euler a", "Heun", "DDIM"],
|
| 127 |
+
info="Sampling method",
|
| 128 |
),
|
| 129 |
],
|
| 130 |
outputs=gr.Image(label="Generated Image"),
|