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)
|