Spaces:
Running
Running
Update app.py
Browse files
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[
|
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[
|
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:
|