pva22
commited on
Commit
·
6c307e5
1
Parent(s):
e693d4a
zero gpu usage activate
Browse files
app.py
CHANGED
|
@@ -154,5 +154,48 @@ with gr.Blocks(css=css) as demo:
|
|
| 154 |
|
| 155 |
result = gr.Image(label="Result", show_label=False)
|
| 156 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 157 |
if __name__ == "__main__":
|
| 158 |
demo.launch()
|
|
|
|
| 154 |
|
| 155 |
result = gr.Image(label="Result", show_label=False)
|
| 156 |
|
| 157 |
+
with gr.Accordion("Advanced Settings", open=False):
|
| 158 |
+
|
| 159 |
+
model_repo_id = gr.Dropdown(
|
| 160 |
+
label="Model Id",
|
| 161 |
+
choices=model_dropdown,
|
| 162 |
+
info="Choose model",
|
| 163 |
+
visible=True,
|
| 164 |
+
allow_custom_value=True,
|
| 165 |
+
value=model_id_default,
|
| 166 |
+
)
|
| 167 |
+
|
| 168 |
+
negative_prompt = gr.Text(
|
| 169 |
+
label="Negative prompt",
|
| 170 |
+
max_lines=1,
|
| 171 |
+
placeholder="Enter a negative prompt",
|
| 172 |
+
visible=True,
|
| 173 |
+
)
|
| 174 |
+
|
| 175 |
+
seed = gr.Slider(
|
| 176 |
+
label="Seed",
|
| 177 |
+
minimum=0,
|
| 178 |
+
maximum=MAX_SEED,
|
| 179 |
+
step=1,
|
| 180 |
+
value=42,
|
| 181 |
+
)
|
| 182 |
+
|
| 183 |
+
randomize_seed = gr.Checkbox(label="Randomize seed", value=False)
|
| 184 |
+
|
| 185 |
+
gr.Examples(examples=examples, inputs=[prompt])
|
| 186 |
+
gr.on(
|
| 187 |
+
triggers=[run_button.click, prompt.submit],
|
| 188 |
+
fn=infer,
|
| 189 |
+
inputs=[
|
| 190 |
+
prompt,
|
| 191 |
+
negative_prompt,
|
| 192 |
+
randomize_seed,
|
| 193 |
+
model_repo_id,
|
| 194 |
+
seed
|
| 195 |
+
],
|
| 196 |
+
outputs=[result, seed],
|
| 197 |
+
)
|
| 198 |
+
|
| 199 |
+
|
| 200 |
if __name__ == "__main__":
|
| 201 |
demo.launch()
|