= commited on
Commit ·
6fc9468
1
Parent(s): d11f700
change
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ MAX_TEXT_CHARS = 512
|
|
| 21 |
@lru_cache(maxsize=1)
|
| 22 |
def get_pipeline():
|
| 23 |
model = AutoModelForTokenClassification.from_pretrained(MODEL_CHECKPOINT)
|
| 24 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
| 25 |
return pipeline(
|
| 26 |
"token-classification",
|
| 27 |
model=model,
|
|
|
|
| 21 |
@lru_cache(maxsize=1)
|
| 22 |
def get_pipeline():
|
| 23 |
model = AutoModelForTokenClassification.from_pretrained(MODEL_CHECKPOINT)
|
| 24 |
+
tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
|
| 25 |
return pipeline(
|
| 26 |
"token-classification",
|
| 27 |
model=model,
|