nlpblogs commited on
Commit
4e4f827
·
verified ·
1 Parent(s): eddfa20

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -102,8 +102,8 @@ if uploaded_files:
102
  for page in pdf_reader.pages:
103
  text_data += page.extract_text()
104
  model = GLiNER.from_pretrained("urchade/gliner_base")
105
- labels = ["person", "country", "organization", "time", "role"]
106
- entities = model.predict_entities(text_data, labels)
107
 
108
  entity_dict = {}
109
  for label in labels:
 
102
  for page in pdf_reader.pages:
103
  text_data += page.extract_text()
104
  model = GLiNER.from_pretrained("urchade/gliner_base")
105
+ labels = ["person", "country", "organization", "time", "role"]
106
+ entities = model.predict_entities(text_data, labels)
107
 
108
  entity_dict = {}
109
  for label in labels: