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:
|