Spaces:

Hasitha16
/

churnsight-ai

Running

Hasitha16 commited on Jul 12

Commit

cd098b8

verified ·

1 Parent(s): dc8cecf

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"
-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

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