Spaces:
Sleeping
Sleeping
Anonymous
commited on
Commit
·
93f8cdd
1
Parent(s):
62f0bbd
add interface
Browse files
app.py
CHANGED
|
@@ -41,7 +41,7 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
|
|
| 41 |
model_512 = load_model_checkpoint(model_512, ckpt_path_512)
|
| 42 |
model_512.eval()
|
| 43 |
model = model_512
|
| 44 |
-
fps =
|
| 45 |
if output_size == "576x1024":
|
| 46 |
width = 1024
|
| 47 |
height = 576
|
|
@@ -85,6 +85,7 @@ def infer(prompt, output_size, seed, num_frames, ddim_steps, unconditional_guida
|
|
| 85 |
model_256.eval()
|
| 86 |
model = model_256
|
| 87 |
fps = 8
|
|
|
|
| 88 |
|
| 89 |
if seed is None:
|
| 90 |
seed = int.from_bytes(os.urandom(2), "big")
|
|
@@ -326,4 +327,4 @@ with gr.Blocks(css=css) as demo:
|
|
| 326 |
outputs=[video_result],
|
| 327 |
api_name="zrscp")
|
| 328 |
|
| 329 |
-
demo.queue(max_size=12).launch(show_api=True)
|
|
|
|
| 41 |
model_512 = load_model_checkpoint(model_512, ckpt_path_512)
|
| 42 |
model_512.eval()
|
| 43 |
model = model_512
|
| 44 |
+
fps = 16
|
| 45 |
if output_size == "576x1024":
|
| 46 |
width = 1024
|
| 47 |
height = 576
|
|
|
|
| 85 |
model_256.eval()
|
| 86 |
model = model_256
|
| 87 |
fps = 8
|
| 88 |
+
print('Model Loaded.')
|
| 89 |
|
| 90 |
if seed is None:
|
| 91 |
seed = int.from_bytes(os.urandom(2), "big")
|
|
|
|
| 327 |
outputs=[video_result],
|
| 328 |
api_name="zrscp")
|
| 329 |
|
| 330 |
+
demo.queue(max_size=12).launch(show_api=True)
|