demo-app / app.py
Girgis711's picture
Update app.py
83301e6
raw
history blame contribute delete
325 Bytes
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)