nlpblogs commited on
Commit
1e10cc9
·
verified ·
1 Parent(s): a518d34

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,7 +64,7 @@ if st.session_state['upload_count'] < max_attempts:
64
  data = pd.Series(text_data, name='Text')
65
  frames = [job, data]
66
  result = pd.concat(frames)
67
- model = pipeline("ner", model="urchade/gliner_base", aggregation_strategy="simple")
68
  labels = ["person", "country", "organization", "date", "time", "role", "skills", "year"]
69
  entities = model(text_data, labels=labels)
70
  df = pd.DataFrame(entities)
@@ -125,7 +125,7 @@ if st.session_state['upload_count'] < max_attempts:
125
  data = pd.Series(text_data, name='Text')
126
  frames = [job, data]
127
  result = pd.concat(frames)
128
- model = pipeline("ner", model="urchade/gliner_base", aggregation_strategy="simple")
129
  labels = ["person", "country", "organization", "date", "time", "role", "skills", "year"]
130
  entities = model(text_data, labels=labels)
131
  df = pd.DataFrame(entities)
 
64
  data = pd.Series(text_data, name='Text')
65
  frames = [job, data]
66
  result = pd.concat(frames)
67
+ model = GLiNER.from_pretrained("urchade/gliner_base")
68
  labels = ["person", "country", "organization", "date", "time", "role", "skills", "year"]
69
  entities = model(text_data, labels=labels)
70
  df = pd.DataFrame(entities)
 
125
  data = pd.Series(text_data, name='Text')
126
  frames = [job, data]
127
  result = pd.concat(frames)
128
+ model = GLiNER.from_pretrained("urchade/gliner_base")
129
  labels = ["person", "country", "organization", "date", "time", "role", "skills", "year"]
130
  entities = model(text_data, labels=labels)
131
  df = pd.DataFrame(entities)