Update app.py
Browse files
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("
|
35 |
-
labels = ["person", "country", "organization"
|
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 = {}
|