idkash1 commited on
Commit
4146450
·
verified ·
1 Parent(s): e05d71b

Update human_text_detect.py

Browse files
Files changed (1) hide show
  1. human_text_detect.py +4 -0
human_text_detect.py CHANGED
@@ -12,6 +12,10 @@ from glob import glob
12
  import spacy
13
  import re
14
  import os
 
 
 
 
15
 
16
  logging.basicConfig(level=logging.INFO)
17
 
 
12
  import spacy
13
  import re
14
  import os
15
+ import gc
16
+
17
+ torch.cuda.empty_cache()
18
+ gc.collect()
19
 
20
  logging.basicConfig(level=logging.INFO)
21