from transformers import pipeline
import torch
import streamlit as st
from random import shuffle

@st.cache_data
def load_sentences():
    with open("sentences.pt", 'rb') as f:
        x = torch.load(f)
        shuffle(x)
    return x

@st.cache_resource
def load_model(model):
    return pipeline("ner",
    model=model,
    aggregation_strategy="average"
    )