Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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(
|
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=[
|
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 |
|