jethrovic commited on
Commit
c57652a
·
1 Parent(s): 621b222

Update app.py

Browse files

changed the model

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
3
 
4
  # Initialize the POS tagger with from_pt=True for PyTorch model
5
  pos_tagger = pipeline(
6
- model="vblagoje/bert-english-uncased-finetuned-pos",
7
  aggregation_strategy="simple",
8
  from_pt=True # Load model from PyTorch weights
9
  )
 
3
 
4
  # Initialize the POS tagger with from_pt=True for PyTorch model
5
  pos_tagger = pipeline(
6
+ model="tomaarsen/span-marker-mbert-base-multinerd",
7
  aggregation_strategy="simple",
8
  from_pt=True # Load model from PyTorch weights
9
  )