Spaces:
Running
on
Zero
Running
on
Zero
amildravid4292
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -373,7 +373,9 @@ def invert(self, image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e
|
|
373 |
|
374 |
|
375 |
@spaces.GPU
|
376 |
-
def run_inversion(
|
|
|
|
|
377 |
init_image = dict["image"].convert("RGB").resize((512, 512))
|
378 |
mask = dict["mask"].convert("RGB").resize((512, 512))
|
379 |
network = invert([init_image], mask, pcs, epochs, weight_decay,lr)
|
@@ -501,9 +503,9 @@ with gr.Blocks(css="style.css") as demo:
|
|
501 |
|
502 |
|
503 |
|
504 |
-
|
505 |
-
|
506 |
-
|
507 |
|
508 |
|
509 |
sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])
|
|
|
373 |
|
374 |
|
375 |
@spaces.GPU
|
376 |
+
def run_inversion(dict, pcs, epochs, weight_decay,lr):
|
377 |
+
print(dict)
|
378 |
+
print(dict.keys())
|
379 |
init_image = dict["image"].convert("RGB").resize((512, 512))
|
380 |
mask = dict["mask"].convert("RGB").resize((512, 512))
|
381 |
network = invert([init_image], mask, pcs, epochs, weight_decay,lr)
|
|
|
503 |
|
504 |
|
505 |
|
506 |
+
invert_button.click(fn=run_inversion,
|
507 |
+
inputs=[input_image, pcs, epochs, weight_decay,lr],
|
508 |
+
outputs = [input_image, file_output])
|
509 |
|
510 |
|
511 |
sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])
|