Linoy Tsaban commited on
Commit
e8feef3
·
1 Parent(s): db41b3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +33 -12
app.py CHANGED
@@ -274,18 +274,39 @@ with gr.Blocks(css='style.css') as demo:
274
  edit_button = gr.Button("Edit")
275
 
276
  with gr.Accordion("Advanced Options", open=False):
277
- with gr.Row():
278
- with gr.Column():
279
- src_prompt = gr.Textbox(lines=1, label="Source Prompt", interactive=True, placeholder="")
280
- steps = gr.Number(value=100, precision=0, label="Num Diffusion Steps", interactive=True)
281
- src_cfg_scale = gr.Number(value=3.5, label=f"Source Guidance Scale", interactive=True)
282
- seed = gr.Number(value=0, precision=0, label="Seed", interactive=True)
283
- randomize_seed = gr.Checkbox(label='Randomize seed', value=False)
284
- with gr.Column():
285
- skip = gr.Slider(minimum=0, maximum=40, value=36, label="Skip Steps", interactive=True)
286
- tar_cfg_scale = gr.Slider(minimum=7, maximum=18,value=15, label=f"Guidance Scale", interactive=True)
287
- sega_edit_guidance = gr.Slider(value=10, label=f"SEGA Edit Guidance Scale", interactive=True)
288
- warm_up = gr.Textbox(label=f"SEGA Warm-up Steps", interactive=True, placeholder="type #warm-up steps for each concpets (e.g. 2,7,5...")
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
289
 
290
 
291
  # neg_guidance = gr.Checkbox(label="SEGA Negative Guidance")
 
274
  edit_button = gr.Button("Edit")
275
 
276
  with gr.Accordion("Advanced Options", open=False):
277
+ with gr.Tabs() as tabs:
278
+ with gr.TabItem('SEGA Guidance', id=0):
279
+ with gr.Row().style(mobile_collapse=False, equal_height=True):
280
+ edit_1 = gr.Textbox(
281
+ label="Edit Prompt 1",
282
+ show_label=False,
283
+ max_lines=1,
284
+ placeholder="Enter your 1st edit prompt",
285
+ ).style(
286
+ border=(True, False, True, True),
287
+ rounded=(True, False, False, True),
288
+ container=False,
289
+ )
290
+ with gr.Group():
291
+ with gr.Row().style(mobile_collapse=False, equal_height=True):
292
+ rev_1 = gr.Checkbox(
293
+ label='Negative Guidance')
294
+ warmup_1 = gr.Slider(label='Warmup', minimum=0, maximum=50, value=10, step=1, interactive=True)
295
+ scale_1 = gr.Slider(label='Scale', minimum=1, maximum=10, value=5, step=0.25, interactive=True)
296
+ threshold_1 = gr.Slider(label='Threshold', minimum=0.5, maximum=0.99, value=0.95, steps=0.01, interactive=True)
297
+ with gr.TabItem('DDPM Guidance', id=0):
298
+ with gr.Row():
299
+ with gr.Column():
300
+ src_prompt = gr.Textbox(lines=1, label="Source Prompt", interactive=True, placeholder="")
301
+ steps = gr.Number(value=100, precision=0, label="Num Diffusion Steps", interactive=True)
302
+ src_cfg_scale = gr.Number(value=3.5, label=f"Source Guidance Scale", interactive=True)
303
+ seed = gr.Number(value=0, precision=0, label="Seed", interactive=True)
304
+ randomize_seed = gr.Checkbox(label='Randomize seed', value=False)
305
+ with gr.Column():
306
+ skip = gr.Slider(minimum=0, maximum=40, value=36, label="Skip Steps", interactive=True)
307
+ tar_cfg_scale = gr.Slider(minimum=7, maximum=18,value=15, label=f"Guidance Scale", interactive=True)
308
+ # sega_edit_guidance = gr.Slider(value=10, label=f"SEGA Edit Guidance Scale", interactive=True)
309
+ # warm_up = gr.Textbox(label=f"SEGA Warm-up Steps", interactive=True, placeholder="type #warm-up steps for each concpets (e.g. 2,7,5...")
310
 
311
 
312
  # neg_guidance = gr.Checkbox(label="SEGA Negative Guidance")