feat: [#535] flush cuda memory

texta_mlp/mlp.py

+import gc
 import logging
 import os
 import pathlib
@@ -310,6 +311,9 @@ class MLP:
             if "sentences" in analyzers and spans == "sentence":
                 document.fact_spans_to_sent()
 
+            gc.collect()
+            torch.cuda.empty_cache()
+
             return document.to_json()
         else:
             return None
@@ -393,4 +397,7 @@ class MLP:
                         result = doc.document_to_json(use_default_doc_path=False)
                         docs[actual_index] = result
 
+        gc.collect()
+        torch.cuda.empty_cache()
+
         return docs