Update handler.py
Browse files- 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=
|
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)
|