Format & polish training and webui

webui.py

@@ -1,3 +1,5 @@
+# flake8: noqa: E402
+
 import sys, os
 import logging
 
@@ -56,20 +58,11 @@ def get_text(text, language_str, hps):
     else:
         bert = torch.zeros(1024, len(phone))
         ja_bert = torch.zeros(768, len(phone))
-    assert bert.shape[-1] == len(phone), (
-        bert.shape,
-        len(phone),
-        sum(word2ph),
-        p1,
-        p2,
-        t1,
-        t2,
-        pold,
-        pold2,
-        word2ph,
-        text,
-        w2pho,
-    )
+
+    assert bert.shape[-1] == len(
+        phone
+    ), f"Bert seq len {bert.shape[-1]} != {len(phone)}"
+
     phone = torch.LongTensor(phone)
     tone = torch.LongTensor(tone)
     language = torch.LongTensor(language)
@@ -138,7 +131,9 @@ if __name__ == "__main__":
         default="./configs/config.json",
         help="path of your config file",
     )
-    parser.add_argument("--share", default=False, help="make link public")
+    parser.add_argument(
+        "--share", default=False, help="make link public", action="store_true"
+    )
     parser.add_argument(
         "-d", "--debug", action="store_true", help="enable DEBUG-LEVEL log"
     )
@@ -163,7 +158,7 @@ if __name__ == "__main__":
         hps.data.filter_length // 2 + 1,
         hps.train.segment_size // hps.data.hop_length,
         n_speakers=hps.data.n_speakers,
-        **hps.model
+        **hps.model,
     ).to(device)
     _ = net_g.eval()