emanuelaboros commited on
Commit
76805b5
·
verified ·
1 Parent(s): 942ee5a

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -90,7 +90,7 @@ class NelPipeline:
90
  print(f"Loading {model_dir}")
91
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
92
  self.tokenizer = AutoTokenizer.from_pretrained(model_dir)
93
- self.model = pipeline("generic-nel", model=NEL_MODEL,
94
  tokenizer=self.tokenizer,
95
  trust_remote_code=True,
96
  device=self.device)
 
90
  print(f"Loading {model_dir}")
91
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
92
  self.tokenizer = AutoTokenizer.from_pretrained(model_dir)
93
+ self.model = pipeline("generic-nel", model="impresso-project/nel-mgenre-multilingual",
94
  tokenizer=self.tokenizer,
95
  trust_remote_code=True,
96
  device=self.device)