Spaces:
Running
on
Zero
Running
on
Zero
MohamedTalaat91
commited on
Commit
•
49ce5f5
1
Parent(s):
1f421f3
Update app.py
Browse files
app.py
CHANGED
@@ -198,6 +198,9 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
|
|
198 |
prompt_name = gr.Dropdown(label="Prompt", choices=PROMPTS_NAMES , value=DEFAULT_PROMPT_NAME)
|
199 |
style_name = gr.Dropdown(label="Style", choices=STYLES_NAMES , value=DEFAULT_STYLE_NAME)
|
200 |
id_image = gr.Image(label="ID Image")
|
|
|
|
|
|
|
201 |
id_weight = gr.Slider(0.0, 3.0, 1, step=0.05, label="id weight")
|
202 |
|
203 |
width = gr.Slider(256, 1536, 896, step=16, label="Width")
|
@@ -216,7 +219,7 @@ def create_demo(args, model_name: str, device: str = "cuda" if torch.cuda.is_ava
|
|
216 |
true_cfg = gr.Slider(1.0, 10.0, 1, step=0.1, label="true CFG scale")
|
217 |
timestep_to_start_cfg = gr.Slider(0, 20, 1, step=1, label="timestep to start cfg", visible=args.dev)
|
218 |
|
219 |
-
|
220 |
|
221 |
with gr.Column():
|
222 |
output_image = gr.Image(label="Generated Image", format='png')
|
|
|
198 |
prompt_name = gr.Dropdown(label="Prompt", choices=PROMPTS_NAMES , value=DEFAULT_PROMPT_NAME)
|
199 |
style_name = gr.Dropdown(label="Style", choices=STYLES_NAMES , value=DEFAULT_STYLE_NAME)
|
200 |
id_image = gr.Image(label="ID Image")
|
201 |
+
|
202 |
+
generate_btn = gr.Button("Generate")
|
203 |
+
|
204 |
id_weight = gr.Slider(0.0, 3.0, 1, step=0.05, label="id weight")
|
205 |
|
206 |
width = gr.Slider(256, 1536, 896, step=16, label="Width")
|
|
|
219 |
true_cfg = gr.Slider(1.0, 10.0, 1, step=0.1, label="true CFG scale")
|
220 |
timestep_to_start_cfg = gr.Slider(0, 20, 1, step=1, label="timestep to start cfg", visible=args.dev)
|
221 |
|
222 |
+
|
223 |
|
224 |
with gr.Column():
|
225 |
output_image = gr.Image(label="Generated Image", format='png')
|