Anonymous commited on
Commit
93f8cdd
1 Parent(s): 62f0bbd

add interface

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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 = 12
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)