Feat: download model from hf

This commit is contained in:
litagin02
2024-05-25 19:05:06 +09:00
parent 2274087da4
commit 5af4e2c6ac
2 changed files with 70 additions and 0 deletions

3
app.py
View File

@@ -10,6 +10,7 @@ from gradio_tabs.inference import create_inference_app
from gradio_tabs.merge import create_merge_app
from gradio_tabs.style_vectors import create_style_vectors_app
from gradio_tabs.train import create_train_app
from gradio_tabs.download_tab import create_download_app
from style_bert_vits2.constants import GRADIO_THEME, VERSION
from style_bert_vits2.nlp.japanese import pyopenjtalk_worker
from style_bert_vits2.nlp.japanese.user_dict import update_dict
@@ -51,6 +52,8 @@ with gr.Blocks(theme=GRADIO_THEME) as app:
create_style_vectors_app()
with gr.Tab("マージ"):
create_merge_app(model_holder=model_holder)
with gr.Tab("モデルダウンロード"):
create_download_app()
app.launch(