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) |