Update human_text_detect.py

human_text_detect.py

@@ -99,8 +99,9 @@ def detect_human_text(model_name, topic, text):
     # Init model
     print('Init model')
     lm_name = 'gpt2-xl' if model_name == 'GPT2XL' else 'microsoft/phi-2'
-    tokenizer = AutoTokenizer.from_pretrained(lm_name)
-    model = AutoModelForCausalLM.from_pretrained(lm_name)
+    cache_dir = "/cache/huggingface"
+    tokenizer = AutoTokenizer.from_pretrained(lm_name, cache_dir=cache_dir)
+    model = AutoModelForCausalLM.from_pretrained(lm_name, cache_dir=cache_dir)
     
     print('Init PerplexityEvaluator')
     sentence_detector = PerplexityEvaluator(model, tokenizer)