Feat: each tab executables

2024-06-16 17:56:32 +09:00
parent 048d0b6c42
commit 7e7ac67906
11 changed files with 104 additions and 3 deletions
--- a/gradio_tabs/inference.py
+++ b/gradio_tabs/inference.py
@@ -523,3 +523,15 @@ def create_inference_app(model_holder: TTSModelHolder) -> gr.Blocks:
        )

    return app
+
+
+if __name__ == "__main__":
+    from config import get_path_config
+    import torch
+
+    path_config = get_path_config()
+    assets_root = path_config.assets_root
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    model_holder = TTSModelHolder(assets_root, device)
+    app = create_inference_app(model_holder)
+    app.launch(inbrowser=True)