Update app.py
Browse files
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",
|
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)
|