Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Buckets new
  • Docs
  • Enterprise
  • Pricing
    • Website
      • Tasks
      • HuggingChat
      • Collections
      • Languages
      • Organizations
    • Community
      • Blog
      • Posts
      • Daily Papers
      • Hardware
      • Learn
      • Discord
      • Forum
      • GitHub
    • Solutions
      • Team & Enterprise
      • Hugging Face PRO
      • Enterprise Support
      • Inference Providers
      • Inference Endpoints
      • Storage Buckets

  • Log In
  • Sign Up

Duplicated from  maidalun1020/bce-reranker-base_v1

nitsuai
/
bce-reranker-base_v1

Text Classification
sentence-transformers
PyTorch
Transformers
xlm-roberta
text-embeddings-inference
Model card Files Files and versions
xet
Community
2

Instructions to use nitsuai/bce-reranker-base_v1 with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

  • Libraries
  • sentence-transformers

    How to use nitsuai/bce-reranker-base_v1 with sentence-transformers:

    from sentence_transformers import CrossEncoder
    
    model = CrossEncoder("nitsuai/bce-reranker-base_v1")
    
    query = "Which planet is known as the Red Planet?"
    passages = [
    	"Venus is often called Earth's twin because of its similar size and proximity.",
    	"Mars, known for its reddish appearance, is often referred to as the Red Planet.",
    	"Jupiter, the largest planet in our solar system, has a prominent red spot.",
    	"Saturn, famous for its rings, is sometimes mistaken for the Red Planet."
    ]
    
    scores = model.predict([(query, passage) for passage in passages])
    print(scores)
  • Transformers

    How to use nitsuai/bce-reranker-base_v1 with Transformers:

    # Use a pipeline as a high-level helper
    from transformers import pipeline
    
    pipe = pipeline("text-classification", model="nitsuai/bce-reranker-base_v1")
    # pip install -U transformers accelerate
    # Load model directly
    from transformers import AutoTokenizer, AutoModelForSequenceClassification
    
    tokenizer = AutoTokenizer.from_pretrained("nitsuai/bce-reranker-base_v1")
    model = AutoModelForSequenceClassification.from_pretrained("nitsuai/bce-reranker-base_v1", device_map="auto")
  • Notebooks
  • Google Colab
  • Kaggle
New discussion
Resources
  • PR & discussions documentation
  • Code of Conduct
  • Hub documentation

Adding `safetensors` variant of this model

#2 opened over 1 year ago by
SFconvertbot

Update model metadata to set pipeline tag to the new `text-ranking` and library name to `sentence-transformers`

#1 opened over 1 year ago by
tomaarsen
Company
TOS Privacy About Careers
Website
Models Datasets Spaces Pricing Docs