Linoy Tsaban commited on
Commit
1df825b
1 Parent(s): 7078734

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -108,7 +108,7 @@ with gr.Blocks() as demo:
108
 
109
 
110
  with gr.Row():
111
- input_image = gr.Image(label="Input Image",interactive=False)
112
  ddpm_edited_image = gr.Image(label=f"Reconstructed Image", interactive=False)
113
  sega_edited_image = gr.Image(label=f"Edited Image", interactive=False)
114
  input_image.style(height=512, width=512)
@@ -147,12 +147,16 @@ with gr.Blocks() as demo:
147
  generate_button.click(
148
  fn=edit,
149
  inputs=[input_image,
150
- src_prompt,
151
- tar_prompt,
152
- steps,
153
- # src_cfg_scale,
154
- skip,
155
- tar_cfg_scale
 
 
 
 
156
  ],
157
  outputs=[ddpm_edited_image],
158
  )
 
108
 
109
 
110
  with gr.Row():
111
+ input_image = gr.Image(label="Input Image")
112
  ddpm_edited_image = gr.Image(label=f"Reconstructed Image", interactive=False)
113
  sega_edited_image = gr.Image(label=f"Edited Image", interactive=False)
114
  input_image.style(height=512, width=512)
 
147
  generate_button.click(
148
  fn=edit,
149
  inputs=[input_image,
150
+ src_prompt,
151
+ tar_prompt,
152
+ steps,
153
+ # src_cfg_scale,
154
+ skip,
155
+ tar_cfg_scale,
156
+ edit_concept,
157
+ sega_edit_guidance,
158
+ warm_up,
159
+ neg_guidance
160
  ],
161
  outputs=[ddpm_edited_image],
162
  )