florian-hoenicke's picture
feat: push custom model
13fae61 verified
|
raw
history blame
1.08 kB
metadata
license: apache-2.0
datasets:
  - fine-tuned/jina-embeddings-v2-base-en-14052024-afuz-webapp
  - allenai/c4
language:
  - en
pipeline_tag: feature-extraction
tags:
  - sentence-transformers
  - feature-extraction
  - sentence-similarity
  - mteb
  - Fantasy
  - Novels
  - Books
  - Fiction
  - Literature

This model is a fine-tuned version of jinaai/jina-embeddings-v2-base-en designed for the following use case:

genre-specific search for fantasy novels

How to Use

This model can be easily integrated into your NLP pipeline for tasks such as text classification, sentiment analysis, entity recognition, and more. Here's a simple example to get you started:

from sentence_transformers import SentenceTransformer
from sentence_transformers.util import cos_sim

model = SentenceTransformer(
    'fine-tuned/jina-embeddings-v2-base-en-14052024-afuz-webapp',
    trust_remote_code=True
)

embeddings = model.encode([
    'first text to embed',
    'second text to embed'
])
print(cos_sim(embeddings[0], embeddings[1]))