Nymbo commited on
Commit
3d1f292
·
verified ·
1 Parent(s): 3d9c7b8

Update Modules/Generate_Image.py

Browse files
Files changed (1) hide show
  1. Modules/Generate_Image.py +8 -6
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"),