Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,6 @@ import json
|
|
10 |
st.set_page_config(layout="wide")
|
11 |
|
12 |
|
13 |
-
|
14 |
model = AutoModelForTokenClassification.from_pretrained("models/lusa")
|
15 |
tokenizer = AutoTokenizer.from_pretrained("models/lusa", model_max_length=512)
|
16 |
tagger = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy='first') #aggregation_strategy='max'
|
|
|
10 |
st.set_page_config(layout="wide")
|
11 |
|
12 |
|
|
|
13 |
model = AutoModelForTokenClassification.from_pretrained("models/lusa")
|
14 |
tokenizer = AutoTokenizer.from_pretrained("models/lusa", model_max_length=512)
|
15 |
tagger = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy='first') #aggregation_strategy='max'
|