IwanSumpena commited on
Commit
a088ce1
·
verified ·
1 Parent(s): 44d3f5e
Files changed (1) hide show
  1. app.py +9 -5
app.py CHANGED
@@ -1,9 +1,13 @@
1
  import streamlit as st
2
- from transformers import pipeline
3
 
4
- pipe=pipeline('sentiment-analysis')
 
 
5
  text=st.text_area('enter some text!')
6
 
7
- if(text):
8
- out =pipe(text)
9
- st.json(out)
 
 
 
 
1
  import streamlit as st
 
2
 
3
+ from gliner import GLiNER
4
+
5
+ model = GLiNER.from_pretrained("urchade/gliner_multi")
6
  text=st.text_area('enter some text!')
7
 
8
+ labels = ["person", "award", "date", "competitions", "teams"]
9
+
10
+ entities = model.predict_entities(text, labels)
11
+
12
+ for entity in entities:
13
+ print(entity["text"], "=>", entity["label"])