Update app.py
Browse files
app.py
CHANGED
@@ -100,6 +100,7 @@ for i in range(1, 51): # Looping for 50 applicants
|
|
100 |
|
101 |
st.divider()
|
102 |
st.subheader("Visualise", divider="blue")
|
|
|
103 |
|
104 |
if 'upload_count' not in st.session_state:
|
105 |
st.session_state['upload_count'] = 0
|
@@ -119,9 +120,9 @@ if st.session_state['upload_count'] < max_attempts:
|
|
119 |
data = pd.Series([text_data], name='Text') # Ensure text_data is also a Series
|
120 |
frames = [job_desc_series, data]
|
121 |
result = pd.concat(frames, ignore_index=True) # Concatenate along rows, reset index
|
122 |
-
|
123 |
labels = ["person", "country", "organization", "role", "skills"]
|
124 |
-
entities =
|
125 |
df = pd.DataFrame(entities)
|
126 |
st.subheader("Applicant's Profile", divider = "orange")
|
127 |
fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
|
|
|
100 |
|
101 |
st.divider()
|
102 |
st.subheader("Visualise", divider="blue")
|
103 |
+
model = SentenceTransformer("all-mpnet-base-v2")
|
104 |
|
105 |
if 'upload_count' not in st.session_state:
|
106 |
st.session_state['upload_count'] = 0
|
|
|
120 |
data = pd.Series([text_data], name='Text') # Ensure text_data is also a Series
|
121 |
frames = [job_desc_series, data]
|
122 |
result = pd.concat(frames, ignore_index=True) # Concatenate along rows, reset index
|
123 |
+
model1 = GLiNER.from_pretrained("urchade/gliner_base")
|
124 |
labels = ["person", "country", "organization", "role", "skills"]
|
125 |
+
entities = model1.predict_entities(text_data, labels)
|
126 |
df = pd.DataFrame(entities)
|
127 |
st.subheader("Applicant's Profile", divider = "orange")
|
128 |
fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
|