Linoy Tsaban commited on
Commit
257ea11
1 Parent(s): a02e0e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -19
app.py CHANGED
@@ -153,15 +153,15 @@ def invert_and_reconstruct(
153
 
154
  def edit(input_image,
155
  wts, zs,
156
- tar_prompt="",
157
- steps=100,
158
- skip=36,
159
- tar_cfg_scale=15,
160
- edit_concepts = [],
161
- guidnace_scales = [],
162
- warmups = [],
163
- neg_guidance = [],
164
- thresholds = []
165
 
166
  ):
167
 
@@ -171,11 +171,11 @@ def edit(input_image,
171
 
172
 
173
  editing_args = dict(
174
- editing_prompt = edit_concepts,
175
- reverse_editing_direction = neg_guidance,
176
- edit_warmup_steps=warmups,
177
- edit_guidance_scale=guidnace_scales,
178
- edit_threshold=thresholds,
179
  edit_momentum_scale=0.5,
180
  edit_mom_beta=0.6,
181
  eta=1,
@@ -343,11 +343,11 @@ with gr.Blocks(css='style.css') as demo:
343
  steps,
344
  skip,
345
  tar_cfg_scale,
346
- [edit_concept_1,edit_concept_2,edit_concept_3],
347
- [guidnace_scale_1,guidnace_scale_2,guidnace_scale_3],
348
- [warmup_1, warmup_2, warmup_3],
349
- [neg_guidance_1, neg_guidance_2, neg_guidance_3],
350
- [threshold_1, threshold_2, threshold_3]
351
 
352
  ],
353
  outputs=[sega_edited_image],
 
153
 
154
  def edit(input_image,
155
  wts, zs,
156
+ tar_prompt,
157
+ steps,
158
+ skipת
159
+ tar_cfg_scale,
160
+ edit_concept_1,edit_concept_2,edit_concept_3,
161
+ guidnace_scale_1,guidnace_scale_2,guidnace_scale_3,
162
+ warmup_1, warmup_2, warmup_3,
163
+ neg_guidance_1, neg_guidance_2, neg_guidance_3,
164
+ threshold_1, threshold_2, threshold_3
165
 
166
  ):
167
 
 
171
 
172
 
173
  editing_args = dict(
174
+ editing_prompt = [edit_concept_1,edit_concept_2,edit_concept_3],
175
+ reverse_editing_direction = [ neg_guidance_1, neg_guidance_2, neg_guidance_3,],
176
+ edit_warmup_steps=[warmup_1, warmup_2, warmup_3,],
177
+ edit_guidance_scale=[guidnace_scale_1,guidnace_scale_2,guidnace_scale_3],
178
+ edit_threshold=[threshold_1, threshold_2, threshold_3],
179
  edit_momentum_scale=0.5,
180
  edit_mom_beta=0.6,
181
  eta=1,
 
343
  steps,
344
  skip,
345
  tar_cfg_scale,
346
+ edit_concept_1,edit_concept_2,edit_concept_3,
347
+ guidnace_scale_1,guidnace_scale_2,guidnace_scale_3,
348
+ warmup_1, warmup_2, warmup_3,
349
+ neg_guidance_1, neg_guidance_2, neg_guidance_3,
350
+ threshold_1, threshold_2, threshold_3
351
 
352
  ],
353
  outputs=[sega_edited_image],