INNER CODE UNIT · Python

load_model

liyucheng09/Selective_Context · app/app.py:265

def load_model(lang):
    model = SelectiveContext(lang=lang)
    return model

if st.button("Compress"):
    model = load_model(lang)
    context, masked_sents = model(text, reduce_ratio=ratio, reduce_level=reduce_level)
    st.subheader("The compressed context is:")
    st.code(context)
    # st.divider()
    st.subheader("The filtered out content is:")
    st.write(masked_sents)

View source record →