Update app.py
Browse files
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 |
-
|
106 |
-
|
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:
|