Instructions to use InfoCoV/Cro-CoV-cseBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use InfoCoV/Cro-CoV-cseBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("fill-mask", model="InfoCoV/Cro-CoV-cseBERT")# Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("InfoCoV/Cro-CoV-cseBERT") model = AutoModelForMaskedLM.from_pretrained("InfoCoV/Cro-CoV-cseBERT") - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Usage:
from sentence_transformers import models
from sentence_transformers import SentenceTransformer
word_embedding_model = models.Transformer('Cro-CoV-cseBERT')
pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(),
pooling_mode_mean_tokens=True,
pooling_mode_cls_token=False,
pooling_mode_max_tokens=False)
model = SentenceTransformer(modules=[word_embedding_model, pooling_model], device='') ## device = 'gpu' or 'cpu'
texts_emb = model.encode(texts)
Datasets:
https://github.com/InfoCoV/InfoCoV
Paper:
Please cite https://www.mdpi.com/2076-3417/11/21/10442
- Downloads last month
- 7