nlpblogs commited on
Commit
b2ed45a
·
verified ·
1 Parent(s): 5daceab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- model = GLiNER.from_pretrained("urchade/gliner_base")
123
  labels = ["person", "country", "organization", "role", "skills"]
124
- entities = model.predict_entities(text_data, labels)
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'],