From 16ed2e13d5370f94918b20662cc61cb6694bcb87 Mon Sep 17 00:00:00 2001
From: Brandon Thomas <echelon@gmail.com>
Date: Sun, 9 Apr 2023 06:49:02 -0400
Subject: [PATCH] Hack for forcing nltk to use our cache

---
 tts/vits/infer_ts_job.py | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/tts/vits/infer_ts_job.py b/tts/vits/infer_ts_job.py
index 95a5b47..394d6d8 100644
--- a/tts/vits/infer_ts_job.py
+++ b/tts/vits/infer_ts_job.py
@@ -2,6 +2,17 @@ import os
 import json
 import argparse
 
+# Adjust NLTK paths
+nltk_cache = os.environ.get('NLTK_DATA', None)
+if nltk_cache:
+    from pathlib import Path
+    Path(nltk_cache).mkdir(parents=True, exist_ok=True)
+    import nltk
+    nltk.data.path = [nltk_cache]
+
+#import nltk
+#print(nltk.data.path)
+
 import torch
 import numpy as np
 import soundfile as sf