Develop

requirements.txt

@@ -1,9 +1,9 @@
-ftfy
-nltk
-emoji
+ftfy==6.1.1
+nltk==3.8.1
+emoji==2.8.0
 torch
-Unidecode
-transformers
-beautifulsoup4
-presidio_anonymizer
+Unidecode==1.3.6
+transformers==4.30.0
+beautifulsoup4==4.12.2
+presidio_anonymizer==2.2.355
 lingua-language-detector

sct/config.py

@@ -52,5 +52,4 @@
 NER_MODELS_LIST = ["FacebookAI/xlm-roberta-large-finetuned-conll03-english", 
               "FacebookAI/xlm-roberta-large-finetuned-conll02-dutch",
               "FacebookAI/xlm-roberta-large-finetuned-conll03-german",
-              "FacebookAI/xlm-roberta-large-finetuned-conll03-spanish",
-              "Babelscape/wikineural-multilingual-ner"]
+              "FacebookAI/xlm-roberta-large-finetuned-conll02-spanish",

sct/utils/ner.py

@@ -36,7 +36,7 @@ def __init__(self):
             model_name = ["FacebookAI/xlm-roberta-large-finetuned-conll03-english",
                           "FacebookAI/xlm-roberta-large-finetuned-conll02-dutch",
                         "FacebookAI/xlm-roberta-large-finetuned-conll03-german",
-                        "FacebookAI/xlm-roberta-large-finetuned-conll03-spanish",
+                        "FacebookAI/xlm-roberta-large-finetuned-conll02-spanish",
                         "Babelscape/wikineural-multilingual-ner"]
 
             english_model_name = model_name[0]

setup.py

@@ -11,14 +11,14 @@
     license='MIT',
     packages=find_packages(),
     install_requires=[
-        'lingua-language-detector>=2.0.0,<2.1',
+        'lingua-language-detector',
         'nltk>=3.8,<3.9',
         'emoji>=2.8,<2.9',
         'ftfy>=6.1,<6.2',
         'Unidecode>=1.3,<1.4',
         'beautifulsoup4>=4.12,<4.13',
         'transformers>=4.30,<4.31',
-        'torch>=2.0,<2.1',
+        'torch',
         'presidio_anonymizer>=2.2.355,<2.3',
     ],
     extras_require={