biobert-tiny-model

作者 yashpatil · 下载 0 · 点赞 0 · 更新 2023-11-01

How to use:

from transformers import AutoModelForMaskedLM, AutoTokenizer, pipeline

# Load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained("yashpatil/biobert-tiny-tokenizer")
model = AutoModelForMaskedLM.from_pretrained("yashpatil/biobert-tiny-model")

unmasker = pipeline('fill-mask', model=model, tokenizer=tokenizer)
output = unmasker("[MASK] are tiny organelles inside cells that are involved in releasing energy from food. This process is known as cellular respiration.")

# Print output
for word in output:
    print(word['token_str'], word['score'])

模型信息

该模型卡暂无预览图和文字说明。以下信息来自 Hugging Face。

作者 yashpatil
类型—
基础模型—
标签—
下载0
点赞0
更新时间2023-11-01
Hugging Face https://huggingface.co/yashpatil/biobert-tiny-model