phillipinseoul
commited on
Commit
·
0b329f4
1
Parent(s):
dff10ea
fix run_button error
Browse files
app.py
CHANGED
@@ -9,12 +9,9 @@ from syncdiffusion.syncdiffusion_model import SyncDiffusion
|
|
9 |
from syncdiffusion.utils import seed_everything
|
10 |
|
11 |
def run_inference(
|
12 |
-
prompt: str,
|
13 |
-
height: int = 512,
|
14 |
width: int = 2048,
|
15 |
sync_weight: float = 20.0,
|
16 |
-
sync_decay_rate: float = 0.95,
|
17 |
-
sync_freq: int = 1,
|
18 |
sync_thres: int = 5,
|
19 |
seed: int = 0
|
20 |
):
|
@@ -81,7 +78,7 @@ if __name__=="__main__":
|
|
81 |
)
|
82 |
|
83 |
ips = [prompt, width, sync_weight, sync_thres, seed]
|
84 |
-
run_button.click(fn=run_inference
|
85 |
|
86 |
demo.queue(max_size=30)
|
87 |
demo.launch()
|
|
|
9 |
from syncdiffusion.utils import seed_everything
|
10 |
|
11 |
def run_inference(
|
12 |
+
prompt: str,
|
|
|
13 |
width: int = 2048,
|
14 |
sync_weight: float = 20.0,
|
|
|
|
|
15 |
sync_thres: int = 5,
|
16 |
seed: int = 0
|
17 |
):
|
|
|
78 |
)
|
79 |
|
80 |
ips = [prompt, width, sync_weight, sync_thres, seed]
|
81 |
+
run_button.click(fn=run_inference, inputs=ips, outputs=[result_gallery])
|
82 |
|
83 |
demo.queue(max_size=30)
|
84 |
demo.launch()
|