Antoni Bigata commited on
Commit
9a75ec8
·
1 Parent(s): e67de88

requirements

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -619,7 +619,7 @@ def sample(
619
 
620
  @spaces.GPU(duration=150)
621
  @torch.no_grad()
622
- def process_video(video_input, audio_input, max_num_seconds):
623
  """Main processing function to generate synchronized video"""
624
 
625
  # Display a message to the user about the processing time
@@ -629,6 +629,8 @@ def process_video(video_input, audio_input, max_num_seconds):
629
  duration=10,
630
  )
631
 
 
 
632
  global \
633
  vae_model, \
634
  hubert_model, \
@@ -1046,7 +1048,7 @@ with gr.Blocks(
1046
  # Show Wordle message when processing starts and hide when complete
1047
  process_button.click(
1048
  fn=process_video,
1049
- inputs=[video_input, audio_input, 10],
1050
  outputs=video_output,
1051
  )
1052
 
 
619
 
620
  @spaces.GPU(duration=150)
621
  @torch.no_grad()
622
+ def process_video(video_input, audio_input):
623
  """Main processing function to generate synchronized video"""
624
 
625
  # Display a message to the user about the processing time
 
629
  duration=10,
630
  )
631
 
632
+ max_num_seconds = 10
633
+
634
  global \
635
  vae_model, \
636
  hubert_model, \
 
1048
  # Show Wordle message when processing starts and hide when complete
1049
  process_button.click(
1050
  fn=process_video,
1051
+ inputs=[video_input, audio_input],
1052
  outputs=video_output,
1053
  )
1054