nlpblogs commited on
Commit
3323972
·
verified ·
1 Parent(s): d9d1a47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", "date", "time", "role", "skills", "year"]
78
  entities = model.predict_entities(text_data, labels)
79
  df = pd.DataFrame(entities)
80
 
81
 
82
- st.subheader("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))
 
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))