Update app.py
Browse files
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 = {}
|