Update human_text_detect.py

human_text_detect.py

@@ -14,9 +14,6 @@ import re
 import os
 import gc
 
-torch.cuda.empty_cache()
-gc.collect()
-
 logging.basicConfig(level=logging.INFO)
 
 def read_all_csv_files(pattern):
@@ -83,6 +80,9 @@ def get_threshold_obj(model_name, topic):
     return threshold
 
 def detect_human_text(model_name, topic, text):
+
+    torch.cuda.empty_cache()
+    gc.collect()
     
     # Get null data
     print('Get null data')