1inkusFace commited on
Commit
0dcd821
·
verified ·
1 Parent(s): 077e79a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -52,12 +52,11 @@ def generate_video(prompt, seed, image=None):
52
  kwargs["image"] = load_image(image=image)
53
  #global predictor
54
  output = predictor.inference(kwargs)
55
- save_dir = f"./result"
56
- os.makedirs(save_dir, exist_ok=True)
57
- video_out_file = f"{save_dir}/{prompt[:100].replace('/','')}_{seed}.mp4"
58
  print(f"generate video, local path: {video_out_file}")
59
  export_to_video(output, video_out_file, fps=24)
60
- return video_out_file, kwargs
61
 
62
  #def create_gradio_interface():
63
  with gr.Blocks() as demo:
@@ -67,11 +66,10 @@ with gr.Blocks() as demo:
67
  seed = gr.Number(label="Random Seed", value=-1)
68
  submit_button = gr.Button("Generate Video")
69
  output_video = gr.Video(label="Generated Video")
70
- output_params = gr.Textbox(label="Output Parameters")
71
  submit_button.click(
72
  fn=generate_video,
73
  inputs=[prompt, seed, image],
74
- outputs=[output_video, output_params],
75
  )
76
  # return demo
77
 
 
52
  kwargs["image"] = load_image(image=image)
53
  #global predictor
54
  output = predictor.inference(kwargs)
55
+ save_dir = f"./"
56
+ video_out_file = f"{save_dir}/{seed}.mp4"
 
57
  print(f"generate video, local path: {video_out_file}")
58
  export_to_video(output, video_out_file, fps=24)
59
+ return video_out_file
60
 
61
  #def create_gradio_interface():
62
  with gr.Blocks() as demo:
 
66
  seed = gr.Number(label="Random Seed", value=-1)
67
  submit_button = gr.Button("Generate Video")
68
  output_video = gr.Video(label="Generated Video")
 
69
  submit_button.click(
70
  fn=generate_video,
71
  inputs=[prompt, seed, image],
72
+ outputs=[output_video],
73
  )
74
  # return demo
75