il-hoon commited on
Commit
612e0ce
·
1 Parent(s): 928b583

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -4,6 +4,8 @@ import shlex
4
  import uuid
5
  import torch
6
 
 
 
7
  import gradio as gr
8
 
9
  def toggle(choice):
@@ -27,7 +29,7 @@ with gr.Blocks() as demo:
27
 
28
 
29
  btn = gr.Button("Generate")
30
- btn.click(fn=inference,inputs=[audio_file, text_input, mic_input], outputs=audio_output)
31
  radio.change(toggle, radio, [mic_input, audio_file])
32
 
33
  demo.launch(enable_queue=True)
 
4
  import uuid
5
  import torch
6
 
7
+ cpu_param = "--cpu" if not torch.cuda.is_available() else ""
8
+
9
  import gradio as gr
10
 
11
  def toggle(choice):
 
29
 
30
 
31
  btn = gr.Button("Generate")
32
+ btn.click(fn=inference.py,inputs=[audio_file, text_input, mic_input], outputs=audio_output)
33
  radio.change(toggle, radio, [mic_input, audio_file])
34
 
35
  demo.launch(enable_queue=True)