Update human_text_detect.py
Browse files- human_text_detect.py +1 -1
human_text_detect.py
CHANGED
|
@@ -96,7 +96,7 @@ def detect_human_text(model_name, topic, text):
|
|
| 96 |
min_tokens_per_sentence = 10
|
| 97 |
max_tokens_per_sentence = 100
|
| 98 |
|
| 99 |
-
cache_dir = f"
|
| 100 |
|
| 101 |
# Init model
|
| 102 |
print('Init model')
|
|
|
|
| 96 |
min_tokens_per_sentence = 10
|
| 97 |
max_tokens_per_sentence = 100
|
| 98 |
|
| 99 |
+
cache_dir = f"./cache/huggingface/{model_name}"
|
| 100 |
|
| 101 |
# Init model
|
| 102 |
print('Init model')
|