dwb2023 commited on
Commit
af4ca43
·
1 Parent(s): c3af845

update classifier model

Browse files
Files changed (1) hide show
  1. src/analyzer.py +1 -1
src/analyzer.py CHANGED
@@ -19,7 +19,7 @@ class EventAnalyzer:
19
 
20
  # Initialize NLP pipelines
21
  self.ner_pipeline = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
22
- self.classifier = pipeline("zero-shot-classification")
23
 
24
  async def extract_entities(self, text: str) -> Dict[str, List[str]]:
25
  """Extract entities from text using NER pipeline."""
 
19
 
20
  # Initialize NLP pipelines
21
  self.ner_pipeline = pipeline("ner", model="dbmdz/bert-large-cased-finetuned-conll03-english")
22
+ self.classifier = pipeline("zero-shot-classification", model="facebook/bart-large-mnli")
23
 
24
  async def extract_entities(self, text: str) -> Dict[str, List[str]]:
25
  """Extract entities from text using NER pipeline."""