Update app.py
Browse files
app.py
CHANGED
@@ -74,12 +74,12 @@ for uploaded_file in uploaded_files:
|
|
74 |
|
75 |
|
76 |
model = GLiNER.from_pretrained("urchade/gliner_base")
|
77 |
-
labels = ["person", "country","organization", "
|
78 |
entities = model.predict_entities(text_data, labels)
|
79 |
df = pd.DataFrame(entities)
|
80 |
|
81 |
|
82 |
-
st.
|
83 |
fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
|
84 |
values='score', color='label')
|
85 |
fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
|
|
|
74 |
|
75 |
|
76 |
model = GLiNER.from_pretrained("urchade/gliner_base")
|
77 |
+
labels = ["person", "country","organization", "role", "skills", "year"]
|
78 |
entities = model.predict_entities(text_data, labels)
|
79 |
df = pd.DataFrame(entities)
|
80 |
|
81 |
|
82 |
+
st.title("Profile of candidate 1")
|
83 |
fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
|
84 |
values='score', color='label')
|
85 |
fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
|