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

Update human_text_detect.py

Browse files
Files changed (1) hide show
  1. human_text_detect.py +5 -3
human_text_detect.py CHANGED
@@ -93,10 +93,12 @@ def detect_human_text(model_name, topic, text):
93
  min_tokens_per_sentence = 10
94
  max_tokens_per_sentence = 100
95
 
96
- cache_dir = f"/cacheHuggingface/{model_name}"
 
 
97
  # Use a writable directory inside the Hugging Face Space
98
- os.makedirs("/cacheHuggingface/PHI2", exist_ok=True)
99
- os.makedirs("/cacheHuggingface/GPT2XL", exist_ok=True)
100
 
101
  # Init model
102
  print('Init model')
 
93
  min_tokens_per_sentence = 10
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)
101
+ os.makedirs("/tmp/cacheHuggingface/GPT2XL", exist_ok=True)
102
 
103
  # Init model
104
  print('Init model')