Update app.py
Browse files
app.py
CHANGED
|
@@ -110,91 +110,93 @@ col-container {
|
|
| 110 |
"""
|
| 111 |
|
| 112 |
with gr.Blocks(css=css) as demo:
|
| 113 |
-
|
| 114 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 119 |
-
|
| 120 |
-
|
| 121 |
-
|
| 122 |
-
|
| 123 |
-
|
| 124 |
-
|
| 125 |
-
|
| 126 |
-
|
| 127 |
-
|
| 128 |
-
|
| 129 |
-
|
| 130 |
-
|
| 131 |
-
|
| 132 |
-
|
| 133 |
-
|
| 134 |
-
|
| 135 |
-
seed = gr.Slider(
|
| 136 |
-
label="Seed",
|
| 137 |
-
minimum=0,
|
| 138 |
-
maximum=MAX_SEED,
|
| 139 |
-
step=1,
|
| 140 |
-
value=0,
|
| 141 |
-
)
|
| 142 |
-
|
| 143 |
-
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
| 144 |
-
|
| 145 |
-
with gr.Row():
|
| 146 |
-
|
| 147 |
-
|
| 148 |
-
|
| 149 |
-
|
| 150 |
-
|
| 151 |
-
|
| 152 |
-
|
| 153 |
-
|
| 154 |
-
|
| 155 |
-
|
| 156 |
-
|
| 157 |
-
|
| 158 |
-
|
| 159 |
-
|
| 160 |
-
|
| 161 |
-
|
| 162 |
-
|
| 163 |
-
|
| 164 |
-
|
| 165 |
-
|
| 166 |
-
|
| 167 |
-
|
| 168 |
-
|
| 169 |
-
|
| 170 |
-
|
| 171 |
-
|
| 172 |
-
|
| 173 |
-
|
| 174 |
-
|
| 175 |
-
|
| 176 |
-
|
| 177 |
-
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
-
|
| 188 |
-
|
| 189 |
-
|
| 190 |
-
|
| 191 |
-
|
| 192 |
-
|
| 193 |
-
|
| 194 |
-
|
| 195 |
-
|
| 196 |
-
|
| 197 |
-
|
| 198 |
-
|
| 199 |
-
|
| 200 |
-
|
|
|
|
|
|
|
|
|
| 110 |
"""
|
| 111 |
|
| 112 |
with gr.Blocks(css=css) as demo:
|
| 113 |
+
with gr.Column(elem_id="col-container"):
|
| 114 |
+
gr.Markdown(f"""
|
| 115 |
+
Generate an image with Flux. Try it out and let me know what you think!
|
| 116 |
+
Expect roughly 45-60 seconds per generation with it's current backend.
|
| 117 |
+
This can be scaled up over time as needed. Thanks!
|
| 118 |
+
Not for Commercial Use - Apache 2.0 License
|
| 119 |
+
""")
|
| 120 |
+
|
| 121 |
+
with gr.Row():
|
| 122 |
+
prompt = gr.Text(
|
| 123 |
+
label="Prompt",
|
| 124 |
+
show_label=False,
|
| 125 |
+
max_lines=1,
|
| 126 |
+
placeholder="Enter your prompt",
|
| 127 |
+
container=False,
|
| 128 |
+
)
|
| 129 |
+
|
| 130 |
+
run_button = gr.Button("Run", scale=0)
|
| 131 |
+
|
| 132 |
+
result = gr.Image(label="Result", show_label=False)
|
| 133 |
+
|
| 134 |
+
with gr.Accordion("Advanced Settings", open=False):
|
| 135 |
+
seed = gr.Slider(
|
| 136 |
+
label="Seed",
|
| 137 |
+
minimum=0,
|
| 138 |
+
maximum=MAX_SEED,
|
| 139 |
+
step=1,
|
| 140 |
+
value=0,
|
| 141 |
+
)
|
| 142 |
+
|
| 143 |
+
randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
|
| 144 |
+
|
| 145 |
+
with gr.Row():
|
| 146 |
+
width = gr.Slider(
|
| 147 |
+
label="Width",
|
| 148 |
+
minimum=256,
|
| 149 |
+
maximum=MAX_IMAGE_SIZE,
|
| 150 |
+
step=32,
|
| 151 |
+
value=1024,
|
| 152 |
+
)
|
| 153 |
+
|
| 154 |
+
height = gr.Slider(
|
| 155 |
+
label="Height",
|
| 156 |
+
minimum=256,
|
| 157 |
+
maximum=MAX_IMAGE_SIZE,
|
| 158 |
+
step=32,
|
| 159 |
+
value=1024,
|
| 160 |
+
)
|
| 161 |
+
|
| 162 |
+
with gr.Row():
|
| 163 |
+
guidance_scale = gr.Slider(
|
| 164 |
+
label="Guidance Scale",
|
| 165 |
+
minimum=1,
|
| 166 |
+
maximum=15,
|
| 167 |
+
step=0.1,
|
| 168 |
+
value=3.5,
|
| 169 |
+
)
|
| 170 |
+
|
| 171 |
+
num_inference_steps = gr.Slider(
|
| 172 |
+
label="Number of inference steps",
|
| 173 |
+
minimum=1,
|
| 174 |
+
maximum=50,
|
| 175 |
+
step=1,
|
| 176 |
+
value=28,
|
| 177 |
+
)
|
| 178 |
+
|
| 179 |
+
gr.Examples(
|
| 180 |
+
examples=examples,
|
| 181 |
+
fn=infer,
|
| 182 |
+
inputs=[prompt],
|
| 183 |
+
outputs=[result, seed],
|
| 184 |
+
cache_examples="lazy",
|
| 185 |
+
)
|
| 186 |
+
|
| 187 |
+
gr.on(
|
| 188 |
+
triggers=[run_button.click, prompt.submit],
|
| 189 |
+
fn=infer,
|
| 190 |
+
inputs=[
|
| 191 |
+
prompt,
|
| 192 |
+
seed,
|
| 193 |
+
randomize_seed,
|
| 194 |
+
width,
|
| 195 |
+
height,
|
| 196 |
+
guidance_scale,
|
| 197 |
+
num_inference_steps,
|
| 198 |
+
],
|
| 199 |
+
outputs=[result, seed],
|
| 200 |
+
)
|
| 201 |
+
|
| 202 |
+
demo.launch()
|