Hugging Face's logo

Spaces:
lfcc
/
Event-Identifier
Sleeping

App Files Community
lfcc commited on
Commit
689cea0
·
1 Parent(s): 7701397

teste

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ 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'
17
 
 
11
 
12
 
13
 
14
+ model = AutoModelForTokenClassification.from_pretrained("models/lusa",use_safetensors=True)
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'
17