NeuralFalcon commited on
Commit
e8a1dbf
·
verified ·
1 Parent(s): 2386301

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -5,6 +5,8 @@ sys.path.append('.')
5
  import torch
6
  import gc
7
  print("Loading model...")
 
 
8
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
9
  print(f'Using device: {device}')
10
  MODEL = build_model('./KOKORO/kokoro-v0_19.pth', device)
@@ -19,7 +21,7 @@ def tts_maker(text,voice_name="af_bella",speed = 0.8,trim=0,pad_between=0,save_p
19
 
20
 
21
  model_list = ["kokoro-v0_19.pth", "kokoro-v0_19-half.pth"]
22
- current_model = model_list[0]
23
 
24
  def update_model(model_name):
25
  """
@@ -110,7 +112,7 @@ with gr.Blocks() as demo1:
110
  with gr.Row():
111
  generate_btn = gr.Button('Generate', variant='primary')
112
  with gr.Accordion('Audio Settings', open=False):
113
- model_name=gr.Dropdown(model_list,label="Model",value=model_list[0])
114
  remove_silence = gr.Checkbox(value=False, label='✂️ Remove Silence From TTS')
115
  minimum_silence = gr.Number(
116
  label="Keep Silence Upto (In seconds)",
@@ -146,7 +148,7 @@ with gr.Blocks() as demo1:
146
  outputs=[audio]
147
  )
148
 
149
- def podcast_maker(text,remove_silence=False,minimum_silence=50,model_name="kokoro-v0_19.pth"):
150
  global MODEL,device
151
  update_model(model_name)
152
  if not minimum_silence:
 
5
  import torch
6
  import gc
7
  print("Loading model...")
8
+ import os
9
+ os.system("python download_model.py")
10
  device = 'cuda' if torch.cuda.is_available() else 'cpu'
11
  print(f'Using device: {device}')
12
  MODEL = build_model('./KOKORO/kokoro-v0_19.pth', device)
 
21
 
22
 
23
  model_list = ["kokoro-v0_19.pth", "kokoro-v0_19-half.pth"]
24
+ current_model = model_list[-1]
25
 
26
  def update_model(model_name):
27
  """
 
112
  with gr.Row():
113
  generate_btn = gr.Button('Generate', variant='primary')
114
  with gr.Accordion('Audio Settings', open=False):
115
+ model_name=gr.Dropdown(model_list,label="Model",value=model_list[-1])
116
  remove_silence = gr.Checkbox(value=False, label='✂️ Remove Silence From TTS')
117
  minimum_silence = gr.Number(
118
  label="Keep Silence Upto (In seconds)",
 
148
  outputs=[audio]
149
  )
150
 
151
+ def podcast_maker(text,remove_silence=False,minimum_silence=50,model_name="kokoro-v0_19-half.pth"):
152
  global MODEL,device
153
  update_model(model_name)
154
  if not minimum_silence: