Views
No views yet
MALWMRLLPLLALLALWGPDPAAAFVNQHLCGSHLVEALYLVCGERGFFYTPKTRREAEDLQVGQVELGGGPGAGSLQPLALEGSLQKRGIVEQCCTSICSLYQLENYCN1from transformers import BertTokenizerFast, BertModel
2checkpoint = 'unikei/bert-base-proteins'
3tokenizer = BertTokenizerFast.from_pretrained(checkpoint)
4model = BertModel.from_pretrained(checkpoint)
5
6example = 'MALWMRLLPLLALLALWGPDPAAAFVNQHLCGSHLVEALYLVCGERGFFYTPKTRREAEDLQVGQVELGGGPGAGSLQPLALEGSLQKRGIVEQCCTSICSLYQLENYCN'
7tokens = tokenizer(example, return_tensors='pt')
8predictions = model(**tokens)