Spaces:

Ehrii
/

sentiment-analysis

Running

JohnDoee commited on Mar 13

Commit

53fd703

1 Parent(s): 2ee1f34

Update main.py

Files changed (1) hide show

main.py CHANGED Viewed

@@ -9,6 +9,7 @@ DetectorFactory.seed = 0
 # Set Hugging Face cache directory to a writable location
 os.environ["HF_HOME"] = "/tmp/huggingface"
 os.makedirs(os.environ["HF_HOME"], exist_ok=True)
 app = FastAPI()

 # Set Hugging Face cache directory to a writable location
 os.environ["HF_HOME"] = "/tmp/huggingface"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
 os.makedirs(os.environ["HF_HOME"], exist_ok=True)
 app = FastAPI()