yerang commited on
Commit
5ec16d2
·
verified ·
1 Parent(s): f912cc3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -56,7 +56,7 @@ audio_path="assets/examples/driving/test_aud.mp3"
56
  #audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
57
 
58
 
59
- @spaces.GPU(duration=120)
60
  def gpu_wrapped_stf_pipeline_execute(audio_path):
61
  return stf_pipeline.execute(audio_path)
62
 
@@ -73,12 +73,13 @@ args = tyro.cli(ArgumentConfig)
73
 
74
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
75
  with gr.Row():
76
- audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
 
77
  stf_button = gr.Button("stf test", variant="primary")
78
  stf_button.click(
79
  fn=gpu_wrapped_stf_pipeline_execute,
80
  inputs=[
81
- audio_path_component
82
  ],
83
  outputs=[driving_video_path]
84
  )
 
56
  #audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
57
 
58
 
59
+ @spaces.GPU()
60
  def gpu_wrapped_stf_pipeline_execute(audio_path):
61
  return stf_pipeline.execute(audio_path)
62
 
 
73
 
74
  with gr.Blocks(theme=gr.themes.Soft()) as demo:
75
  with gr.Row():
76
+ #audio_path_component = gr.Textbox(label="Input", value="assets/examples/driving/test_aud.mp3")
77
+ input_audio = gr.Audio(label="Upload or Record an audio", type="filepath")
78
  stf_button = gr.Button("stf test", variant="primary")
79
  stf_button.click(
80
  fn=gpu_wrapped_stf_pipeline_execute,
81
  inputs=[
82
+ input_audio
83
  ],
84
  outputs=[driving_video_path]
85
  )