Olivier CARON commited on
Commit
d3bad20
1 Parent(s): b25dbae

Update gliner_file.py

Browse files
Files changed (1) hide show
  1. gliner_file.py +1 -1
gliner_file.py CHANGED
@@ -5,7 +5,7 @@ def run_ner(model, text, labels_list, threshold=0.4):
5
  entities = model.predict_entities(text, labels_list, threshold=threshold)
6
 
7
  # Loading the GLiNER model
8
- model = GLiNER.from_pretrained("urchade/gliner_largev2")
9
  model.eval() # Put the model in evaluation mode
10
 
11
  # Initializing the dictionary to store the results
 
5
  entities = model.predict_entities(text, labels_list, threshold=threshold)
6
 
7
  # Loading the GLiNER model
8
+ model = GLiNER.from_pretrained("urchade/gliner_multi-v2.1")
9
  model.eval() # Put the model in evaluation mode
10
 
11
  # Initializing the dictionary to store the results