Spaces:
Paused
Paused
Commit ·
0c19c8e
1
Parent(s): 42b0869
added punkt_tab dependency
Browse files- Dockerfile +2 -1
- 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 |
-
|
|
|
|
| 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):
|