tudorgeorgescu commited on
Commit
2912fb9
·
verified ·
1 Parent(s): 443abf4
Files changed (1) hide show
  1. app.py +29 -0
app.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import pipeline
3
+
4
+ # Load the keyphrase extraction models
5
+ model1 = pipeline("ner", model="ml6team/keyphrase-extraction-kbir-inspec")
6
+ model2 = pipeline("ner", model="aglazkova/bart_finetuned_keyphrase_extraction")
7
+
8
+ # Streamlit app
9
+ st.title("Keyphrase Extraction App")
10
+ st.write("Enter text below to extract keyphrases using two different models:")
11
+
12
+ # Text input
13
+ user_input = st.text_area("Text Input", "")
14
+
15
+ # Extract keyphrases
16
+ if st.button("Extract Keyphrases"):
17
+ if user_input:
18
+ keyphrases_model1 = model1(user_input)
19
+ keyphrases_model2 = model2(user_input)
20
+
21
+ st.write("Extracted Keyphrases from `ml6team/keyphrase-extraction-kbir-inspec`:")
22
+ for keyphrase in keyphrases_model1:
23
+ st.write(keyphrase['word'])
24
+
25
+ st.write("Extracted Keyphrases from `aglazkova/bart_finetuned_keyphrase_extraction`:")
26
+ for keyphrase in keyphrases_model2:
27
+ st.write(keyphrase['word'])
28
+ else:
29
+ st.write("Please enter some text to extract keyphrases.")