Update human_text_detect.py

human_text_detect.py

@@ -96,12 +96,11 @@ def detect_human_text(model_name, topic, text):
     min_tokens_per_sentence = 10
     max_tokens_per_sentence = 100
 
-    cache_dir = f"/cache/huggingface/{model_name}"
+    cache_dir = f"/.cache/huggingface/{model_name}"
     
     # Init model
     print('Init model')
     lm_name = 'gpt2-xl' if model_name == 'GPT2XL' else 'microsoft/phi-2'
-    cache_dir = "/cache/huggingface"
     tokenizer = AutoTokenizer.from_pretrained(lm_name, cache_dir=cache_dir)
     
     print("Save tokenizer")