Anonymous commited on
Commit
f958840
·
1 Parent(s): 07c31ce
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -25,10 +25,10 @@ os.makedirs(ckpt_dir_1024, exist_ok=True)
25
  ckpt_path_256 = "checkpoints/base_256_v1/model.pth"
26
  ckpt_dir_256 = "checkpoints/base_256_v1"
27
  os.makedirs(ckpt_dir_256, exist_ok=True)
28
- hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=ckpt_dir_256)
29
 
30
 
31
- def infer(prompt, output_size):
32
  num_frames = 32
33
  ddim_steps = 50
34
  unconditional_guidance_scale = 12.0
@@ -261,10 +261,10 @@ with gr.Blocks(css=css) as demo:
261
  submit_btn = gr.Button("Generate")
262
  video_result = gr.Video(label="Video Output", elem_id="video-output")
263
 
264
- gr.Examples(examples=examples, inputs=[prompt_in, output_size])
265
 
266
  submit_btn.click(fn=infer,
267
- inputs=[prompt_in, output_size],
268
  outputs=[video_result],
269
  api_name="zrscp")
270
 
 
25
  ckpt_path_256 = "checkpoints/base_256_v1/model.pth"
26
  ckpt_dir_256 = "checkpoints/base_256_v1"
27
  os.makedirs(ckpt_dir_256, exist_ok=True)
28
+ # hf_hub_download(repo_id="MoonQiu/LongerCrafter", filename="model.pth", local_dir=ckpt_dir_256)
29
 
30
 
31
+ def infer(prompt, output_size="256x256"):
32
  num_frames = 32
33
  ddim_steps = 50
34
  unconditional_guidance_scale = 12.0
 
261
  submit_btn = gr.Button("Generate")
262
  video_result = gr.Video(label="Video Output", elem_id="video-output")
263
 
264
+ gr.Examples(examples=examples, inputs=[prompt_in])
265
 
266
  submit_btn.click(fn=infer,
267
+ inputs=[prompt_in],
268
  outputs=[video_result],
269
  api_name="zrscp")
270