app.py

import streamlit as st
from transformers import pipeline

# Load the keyphrase extraction models
model1 = pipeline("ner", model="ml6team/keyphrase-extraction-kbir-inspec")
model2 = pipeline("ner", model="aglazkova/bart_finetuned_keyphrase_extraction")

# Streamlit app
st.title("Keyphrase Extraction App")
st.write("Enter text below to extract keyphrases using two different models:")

# Text input
user_input = st.text_area("Text Input", "")

# Extract keyphrases
if st.button("Extract Keyphrases"):
    if user_input:
        keyphrases_model1 = model1(user_input)
        keyphrases_model2 = model2(user_input)

        st.write("Extracted Keyphrases from `ml6team/keyphrase-extraction-kbir-inspec`:")
        for keyphrase in keyphrases_model1:
            st.write(keyphrase['word'])

        st.write("Extracted Keyphrases from `aglazkova/bart_finetuned_keyphrase_extraction`:")
        for keyphrase in keyphrases_model2:
            st.write(keyphrase['word'])
    else:
        st.write("Please enter some text to extract keyphrases.")