Update analyzers/ner_analyzer.py

analyzers/ner_analyzer.py

@@ -9,15 +9,15 @@ logger = logging.getLogger(__name__)
 
 class NERAnalyzer(BaseAnalyzer):
     def __init__(self):
-        self.model_name = "jpbahiaz/bert-base-portuguese-ner"  # Modelo NER mais leve para português
+        self.model_name = "neuralmind/bert-base-portuguese-cased"  # Modelo NER público para português
         logger.info(f"Carregando o modelo NER: {self.model_name}")
         
-        # Carregando o modelo e tokenizer sem necessidade de token de autenticação
+        # Carregando o modelo e tokenizer
         self.model = AutoModelForTokenClassification.from_pretrained(self.model_name)
         self.tokenizer = AutoTokenizer.from_pretrained(self.model_name)
         
         # Definindo as labels que queremos extrair (pessoas e organizações)
-        self.target_labels = ['B-PESSOA', 'I-PESSOA', 'B-ORGANIZACAO', 'I-ORGANIZACAO']
+        self.target_labels = ['B-PER', 'I-PER', 'B-ORG', 'I-ORG']
         
         logger.info("Modelo NER e tokenizador carregados com sucesso")