nlpblogs commited on
Commit
eea9f45
·
verified ·
1 Parent(s): d00d425

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -90,7 +90,7 @@ if st.session_state['upload_count'] < max_attempts:
90
  result = pd.concat(frames)
91
  st.dataframe(result)
92
  model = GLiNER.from_pretrained("urchade/gliner_base")
93
- labels = ["person", "country", "city", "organization", "date", "money", "percent value", "position"]
94
  entities = model.predict_entities(text_data, labels)
95
  df = pd.DataFrame(entities)
96
  st.dataframe(entities)
 
90
  result = pd.concat(frames)
91
  st.dataframe(result)
92
  model = GLiNER.from_pretrained("urchade/gliner_base")
93
+ labels = ["person", "country","organization", "date", "time", "role", "skills", "year"]
94
  entities = model.predict_entities(text_data, labels)
95
  df = pd.DataFrame(entities)
96
  st.dataframe(entities)