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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -31,8 +31,8 @@ if uploaded_files:
31
  for page in pdf_reader.pages:
32
  text_data += page.extract_text()
33
 
34
- model = GLiNER.from_pretrained("xomad/gliner-model-merge-large-v1.0")
35
- labels = ["person", "country", "organization", "date","position", "skills", "languages"]
36
  entities = model.predict_entities(text_data, labels)
37
 
38
  entity_dict = {}
 
31
  for page in pdf_reader.pages:
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 = {}