Update inference.py

inference.py

@@ -293,6 +293,26 @@ class StyleTTS2(torch.nn.Module):
         
         tokens = [0] + tokens + [0]
         tokens = torch.LongTensor(tokens).unsqueeze(0).to(device)
+        import yaml
+        
+        config_path = "config.yaml"   # <-- đổi thành path config bạn đang dùng
+        
+        with open(config_path, "r", encoding="utf-8") as f:
+            cfg = yaml.safe_load(f)
+        
+        symbols = (
+            list(cfg['symbol']['pad']) +
+            list(cfg['symbol']['punctuation']) +
+            list(cfg['symbol']['letters']) +
+            list(cfg['symbol']['letters_ipa']) +
+            list(cfg['symbol']['extend'])
+        )
+        
+        for i in range(20):
+            print(i, repr(symbols[i]))
+        
+        print("Total:", len(symbols)+1)
+
 
 
         with torch.no_grad():