Downloads · 30 days
11
1% of all-time downloads
InfoCoV/Cro-CoV-cseBERT
Cro-CoV-cseBERT is a fill-mask model from InfoCoV. Use it when you need the model to fill a missing word. It is set up for transformers.
Downloads · 30 days
11
1% of all-time downloads
All-time downloads
915
Public
Repo size
2 GB
Likes
0
Public
Click a slice to open those files.
.pt994 MB · 67%
From the Hugging Face model README
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)
https://github.com/InfoCoV/InfoCoV
Please cite https://www.mdpi.com/2076-3417/11/21/10442