amildravid4292 commited on
Commit
bf723db
·
verified ·
1 Parent(s): 666901c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -153,7 +153,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
153
 
154
  @torch.no_grad()
155
  @spaces.GPU
156
- def edit_inference(network, prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
157
  global device
158
  #global generator
159
  global unet
@@ -485,7 +485,7 @@ with gr.Blocks(css="style.css") as demo:
485
  sample.click(fn=sample_then_run, outputs=[input_image, file_output])
486
 
487
  submit.click(
488
- fn=edit_inference, inputs=[network, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
489
  )
490
  file_input.change(fn=file_upload, inputs=file_input, outputs = gallery)
491
 
 
153
 
154
  @torch.no_grad()
155
  @spaces.GPU
156
+ def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
157
  global device
158
  #global generator
159
  global unet
 
485
  sample.click(fn=sample_then_run, outputs=[input_image, file_output])
486
 
487
  submit.click(
488
+ fn=edit_inference, inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
489
  )
490
  file_input.change(fn=file_upload, inputs=file_input, outputs = gallery)
491