Spaces:
Runtime error
Runtime error
Emanuela Boros
commited on
Commit
•
7e20248
1
Parent(s):
c80d375
Add special pipeline
Browse files
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(
|
|
|
|
|
|
|
|
|
|
|
|
|
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):
|