Spaces:

Hasitha16
/

churnsight-ai

Sleeping

Hasitha16 commited on Jul 12

Commit

dc8cecf

verified ·

1 Parent(s): 80cb61e

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -1,7 +1,8 @@
 import os
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf-cache"
 import nltk
-nltk.download("punkt")
 from typing import List, Optional
 from pydantic import BaseModel
 from transformers import pipeline

 import os
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf-cache"
+os.environ["NLTK_DATA"] = "/tmp/nltk_data"
 import nltk
+nltk.download("punkt", download_dir="/tmp/nltk_data")
 from typing import List, Optional
 from pydantic import BaseModel
 from transformers import pipeline