Emanuela Boros commited on
Commit
7e20248
1 Parent(s): c80d375

Add special pipeline

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -3,7 +3,13 @@ from transformers import pipeline
3
  import os
4
 
5
  token = os.getenv("HF_TOKEN")
6
- get_completion = pipeline("ner", model="impresso-project/bert-newsagency-ner-fr")
 
 
 
 
 
 
7
 
8
 
9
  def ner(input):
 
3
  import os
4
 
5
  token = os.getenv("HF_TOKEN")
6
+ get_completion = pipeline(
7
+ "newsagency-ner",
8
+ model="impresso-project/bert-newsagency-ner-fr",
9
+ trust_remote_code=True,
10
+ revision="main",
11
+ token=token,
12
+ )
13
 
14
 
15
  def ner(input):