delete torch dep

Clearbox-AI · Jan 29, 2024 · fa29091 · fa29091
1 parent e71dafa
commit fa29091
Show file tree

Hide file tree

Showing 4 changed files with 343 additions and 709 deletions.
diff --git a/nerpii/named_entity_recognizer.py b/nerpii/named_entity_recognizer.py
@@ -13,7 +13,14 @@
 )
 from presidio_analyzer.nlp_engine import NlpEngineProvider
 import spacy
-from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
+from transformers import (
+    AutoModelForTokenClassification,
+    AutoTokenizer,
+    logging,
+    pipeline,
+)
+
+logging.set_verbosity_error()
 
 
 # è importante eseguire prima assign_entities_with_presidio, poi