CherithCutestory commited on
Commit
0c19c8e
·
1 Parent(s): 42b0869

added punkt_tab dependency

Browse files
Files changed (2) hide show
  1. Dockerfile +2 -1
  2. app.py +10 -0
Dockerfile CHANGED
@@ -16,7 +16,8 @@ WORKDIR /app
16
  COPY requirements.txt .
17
  RUN pip install --no-cache-dir -r requirements.txt
18
 
19
- RUN python -c "import nltk; nltk.download('punkt'); nltk.download('punkt_tab')" || true
 
20
 
21
  RUN python -c "\
22
  import torch; \
 
16
  COPY requirements.txt .
17
  RUN pip install --no-cache-dir -r requirements.txt
18
 
19
+ ENV NLTK_DATA=/usr/local/share/nltk_data
20
+ RUN python -c "import nltk; nltk.download('punkt', download_dir='/usr/local/share/nltk_data'); nltk.download('punkt_tab', download_dir='/usr/local/share/nltk_data'); nltk.download('averaged_perceptron_tagger_eng', download_dir='/usr/local/share/nltk_data')" || true
21
 
22
  RUN python -c "\
23
  import torch; \
app.py CHANGED
@@ -42,8 +42,18 @@ EMOTION_PRESETS = {
42
  tts_engine = None
43
 
44
 
 
 
 
 
 
 
 
 
 
45
  def load_model():
46
  global tts_engine
 
47
 
48
  _original_load = torch.load
49
  def _patched_load(*args, **kwargs):
 
42
  tts_engine = None
43
 
44
 
45
+ def ensure_nltk_data():
46
+ import nltk
47
+ for pkg in ['punkt', 'punkt_tab', 'averaged_perceptron_tagger_eng']:
48
+ try:
49
+ nltk.data.find(f'tokenizers/{pkg}' if 'punkt' in pkg else f'taggers/{pkg}')
50
+ except LookupError:
51
+ nltk.download(pkg)
52
+
53
+
54
  def load_model():
55
  global tts_engine
56
+ ensure_nltk_data()
57
 
58
  _original_load = torch.load
59
  def _patched_load(*args, **kwargs):