import streamlit as st
from transformers import pipeline

from span_marker import SpanMarkerModel

# Download from the 🤗 Hub
model = SpanMarkerModel.from_pretrained("tomaarsen/span-marker-bert-base-acronyms")
# Run inference
text = st.text_area('enter some text:')

if text:
    out = model.predict(text)

    st.json(out)