Text Retrieval
Transformers
Safetensors
sentence-transformers
English
kpr-bert
feature-extraction
custom_code
Instructions to use knowledgeable-ai/kpr-bge-base-en with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use knowledgeable-ai/kpr-bge-base-en with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("knowledgeable-ai/kpr-bge-base-en", trust_remote_code=True, dtype="auto") - sentence-transformers
How to use knowledgeable-ai/kpr-bge-base-en with sentence-transformers:
from sentence_transformers import SentenceTransformer model = SentenceTransformer("knowledgeable-ai/kpr-bge-base-en", trust_remote_code=True) sentences = [ "The weather is lovely today.", "It's so sunny outside!", "He drove to the stadium." ] embeddings = model.encode(sentences) similarities = model.similarity(embeddings, embeddings) print(similarities.shape) # [3, 3] - Notebooks
- Google Colab
- Kaggle
Add pipeline tag: text-retrieval
#1
by nielsr HF Staff - opened
This PR improves the model card by adding the pipeline_tag: text-retrieval to the metadata. This ensures the model is correctly categorized and more easily discoverable on the Hugging Face Hub for users looking for text retrieval models.
ikuyamada changed pull request status to merged