import streamlit as st
from transformers import pipeline

# Load the NER pipeline
ner_pipe = pipeline("ner", model="rouabelgacem/rouabelgacem-jira-bert-nerr", tokenizer="rouabelgacem/rouabelgacem-jira-bert-nerr")

# Streamlit app
st.title("Jira BERT NER App")

# Input text box
text = st.text_area("Enter Jira-related text:")

if text:
    # Perform Named Entity Recognition
    entities = ner_pipe(text)

    # Display NER result
    st.subheader("Named Entity Recognition Result:")
    st.json(entities)