idkash1 commited on
Commit
ebea54d
·
verified ·
1 Parent(s): b956ba4

Update human_text_detect.py

Browse files
Files changed (1) hide show
  1. human_text_detect.py +2 -1
human_text_detect.py CHANGED
@@ -94,7 +94,8 @@ def detect_human_text(model_name, topic, text):
94
  max_tokens_per_sentence = 100
95
 
96
  cache_dir = f"/tmp/cacheHuggingface/{model_name}"
97
-
 
98
  print('Create dir')
99
  # Use a writable directory inside the Hugging Face Space
100
  os.makedirs("/tmp/cacheHuggingface/PHI2", exist_ok=True)
 
94
  max_tokens_per_sentence = 100
95
 
96
  cache_dir = f"/tmp/cacheHuggingface/{model_name}"
97
+ os.environ["TRANSFORMERS_CACHE"] = cache_dir
98
+
99
  print('Create dir')
100
  # Use a writable directory inside the Hugging Face Space
101
  os.makedirs("/tmp/cacheHuggingface/PHI2", exist_ok=True)