tiezhen HF Staff commited on
Commit
59610d9
·
verified ·
1 Parent(s): 9a39ad9

Update tts/gradio_api.py

Browse files
Files changed (1) hide show
  1. tts/gradio_api.py +2 -3
tts/gradio_api.py CHANGED
@@ -62,7 +62,7 @@ def model_worker(input_queue, output_queue, device_id):
62
  raise gr.Error("Generation failed")
63
 
64
 
65
- def main(inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w, processes, input_queue, output_queue):
66
  input_queue = mp_manager.Queue()
67
  print("Push task to the inp queue |", inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w)
68
  input_queue.put((inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w))
@@ -87,8 +87,7 @@ if __name__ == '__main__':
87
  processes = []
88
 
89
  api_interface = gr.Interface(fn=
90
- partial(main, processes=processes, input_queue=input_queue,
91
- output_queue=output_queue),
92
  inputs=[gr.Audio(type="filepath", label="Upload .wav"), gr.File(type="filepath", label="Upload .npy"), "text",
93
  gr.Number(label="infer timestep", value=32),
94
  gr.Number(label="Intelligibility Weight", value=1.4),
 
62
  raise gr.Error("Generation failed")
63
 
64
 
65
+ def main(inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w, processes):
66
  input_queue = mp_manager.Queue()
67
  print("Push task to the inp queue |", inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w)
68
  input_queue.put((inp_audio, inp_npy, inp_text, infer_timestep, p_w, t_w))
 
87
  processes = []
88
 
89
  api_interface = gr.Interface(fn=
90
+ partial(main, processes=processes),
 
91
  inputs=[gr.Audio(type="filepath", label="Upload .wav"), gr.File(type="filepath", label="Upload .npy"), "text",
92
  gr.Number(label="infer timestep", value=32),
93
  gr.Number(label="Intelligibility Weight", value=1.4),