nlpblogs commited on
Commit
bb8d1f2
·
verified ·
1 Parent(s): 674df9a

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
 
93
- model = pipeline("ner", model="xomad/gliner-model-merge-large-v1.0", aggregation_strategy="simple")
94
  labels = ["person", "country", "city", "organization", "date", "money", "percent value", "position"]
95
  entities = model(text_data, labels=labels)
96
  df_entities = pd.DataFrame(entities)
 
90
  result = pd.concat(frames)
91
  st.dataframe(result)
92
 
93
+ model = GLiNER.from_pretrained("urchade/gliner_base")
94
  labels = ["person", "country", "city", "organization", "date", "money", "percent value", "position"]
95
  entities = model(text_data, labels=labels)
96
  df_entities = pd.DataFrame(entities)