| from transformers import pipeline | |
| import torch | |
| import streamlit as st | |
| from random import shuffle | |
| def load_sentences(): | |
| with open("sentences.pt", 'rb') as f: | |
| x = torch.load(f) | |
| shuffle(x) | |
| return x | |
| def load_model(model): | |
| return pipeline("ner", | |
| model=model, | |
| aggregation_strategy="average" | |
| ) |