nlpblogs commited on
Commit
d6ea9fc
·
verified ·
1 Parent(s): 7a0c57e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ if uploaded_files:
32
  text_data += page.extract_text()
33
 
34
  model = GLiNER.from_pretrained("urchade/gliner_base")
35
- labels = ["person", "country", "organization"]
36
  entities = model.predict_entities(text_data, labels)
37
 
38
  entity_dict = {}
 
32
  text_data += page.extract_text()
33
 
34
  model = GLiNER.from_pretrained("urchade/gliner_base")
35
+ labels = ["person", "country", "organization", "time", "role"]
36
  entities = model.predict_entities(text_data, labels)
37
 
38
  entity_dict = {}