Merge pull request #29 from huggingface/nltk-fix

fix issue with ntlk
2024-08-21 18:13:31 +02:00
parent ba40056ac1 c53fdacb2d
commit ec3c225a0f
1 changed files with 1 additions and 1 deletions
--- a/s2s_pipeline.py
+++ b/s2s_pipeline.py
@@ -40,7 +40,7 @@ from utils import (
 # Ensure that the necessary NLTK resources are available
 try:
    nltk.data.find('tokenizers/punkt_tab')
-except LookupError:
+except (LookupError, OSError):
    nltk.download('punkt_tab')

 # caching allows ~50% compilation time reduction