Spaces:
Running
Running
Update README.md
Browse files
README.md
CHANGED
@@ -1,10 +1,124 @@
|
|
1 |
---
|
2 |
title: README
|
3 |
-
emoji:
|
4 |
-
colorFrom:
|
5 |
-
colorTo:
|
6 |
sdk: static
|
7 |
pinned: false
|
8 |
---
|
|
|
9 |
|
10 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
---
|
2 |
title: README
|
3 |
+
emoji: ❤️
|
4 |
+
colorFrom: red
|
5 |
+
colorTo: red
|
6 |
sdk: static
|
7 |
pinned: false
|
8 |
---
|
9 |
+
SentenceTransformers 🤗 is a Python framework for state-of-the-art sentence, text and image embeddings.
|
10 |
|
11 |
+
Install the [Sentence Transformers](https://www.sbert.net/) library.
|
12 |
+
```
|
13 |
+
pip install -U sentence-transformers
|
14 |
+
```
|
15 |
+
|
16 |
+
The usage is as simple as:
|
17 |
+
```python
|
18 |
+
from sentence_transformers import CrossEncoder
|
19 |
+
|
20 |
+
# Load a pre-trained CrossEncoder model
|
21 |
+
model = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
22 |
+
|
23 |
+
# Predict scores for a pair of sentences
|
24 |
+
scores = model.predict([
|
25 |
+
("How many people live in Berlin?", "Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers."),
|
26 |
+
("How many people live in Berlin?", "Berlin is well known for its museums."),
|
27 |
+
])
|
28 |
+
# => array([ 8.607138 , -4.3200774], dtype=float32)
|
29 |
+
```
|
30 |
+
|
31 |
+
Alternatively, you can also use the `CrossEncoder.rank` argument to rerank documents given a query:
|
32 |
+
```python
|
33 |
+
from sentence_transformers import CrossEncoder
|
34 |
+
|
35 |
+
# Load a pre-trained CrossEncoder model
|
36 |
+
model = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-6-v2")
|
37 |
+
|
38 |
+
# Rank a list of passages for a query
|
39 |
+
query = "How many people live in Berlin?"
|
40 |
+
passages = [
|
41 |
+
"Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.",
|
42 |
+
"Berlin is well known for its museums.",
|
43 |
+
"In 2014, the city state Berlin had 37,368 live births (+6.6%), a record number since 1991.",
|
44 |
+
"The urban area of Berlin comprised about 4.1 million people in 2014, making it the seventh most populous urban area in the European Union.",
|
45 |
+
"The city of Paris had a population of 2,165,423 people within its administrative city limits as of January 1, 2019",
|
46 |
+
"An estimated 300,000-420,000 Muslims reside in Berlin, making up about 8-11 percent of the population.",
|
47 |
+
"Berlin is subdivided into 12 boroughs or districts (Bezirke).",
|
48 |
+
"In 2015, the total labour force in Berlin was 1.85 million.",
|
49 |
+
"In 2013 around 600,000 Berliners were registered in one of the more than 2,300 sport and fitness clubs.",
|
50 |
+
"Berlin has a yearly total of about 135 million day visitors, which puts it in third place among the most-visited city destinations in the European Union.",
|
51 |
+
]
|
52 |
+
ranks = model.rank(query, passages)
|
53 |
+
|
54 |
+
# Print the scores
|
55 |
+
print("Query:", query)
|
56 |
+
for rank in ranks:
|
57 |
+
print(f"{rank['score']:.2f}\t{passages[rank['corpus_id']]}")
|
58 |
+
"""
|
59 |
+
Query: How many people live in Berlin?
|
60 |
+
8.92 The urban area of Berlin comprised about 4.1 million people in 2014, making it the seventh most populous urban area in the European Union.
|
61 |
+
8.61 Berlin had a population of 3,520,031 registered inhabitants in an area of 891.82 square kilometers.
|
62 |
+
8.24 An estimated 300,000-420,000 Muslims reside in Berlin, making up about 8-11 percent of the population.
|
63 |
+
7.60 In 2014, the city state Berlin had 37,368 live births (+6.6%), a record number since 1991.
|
64 |
+
6.35 In 2013 around 600,000 Berliners were registered in one of the more than 2,300 sport and fitness clubs.
|
65 |
+
5.42 Berlin has a yearly total of about 135 million day visitors, which puts it in third place among the most-visited city destinations in the European Union.
|
66 |
+
3.45 In 2015, the total labour force in Berlin was 1.85 million.
|
67 |
+
0.33 Berlin is subdivided into 12 boroughs or districts (Bezirke).
|
68 |
+
-4.24 The city of Paris had a population of 2,165,423 people within its administrative city limits as of January 1, 2019
|
69 |
+
-4.32 Berlin is well known for its museums.
|
70 |
+
"""
|
71 |
+
```
|
72 |
+
|
73 |
+
Hugging Face makes it easy to collaboratively build and showcase your [Sentence Transformers](https://www.sbert.net/) models! You can collaborate with your organization, upload and showcase your own models in your profile ❤️
|
74 |
+
|
75 |
+
<div class="grid lg:grid-cols-2 gap-x-4 gap-y-7">
|
76 |
+
<a href="https://www.sbert.net/" class="block overflow-hidden group">
|
77 |
+
<div
|
78 |
+
class="w-full h-40 object-cover mb-2 rounded-lg flex items-center justify-center bg-[#FA8072]"
|
79 |
+
>
|
80 |
+
<img alt="" src="https://huggingface.co/spaces/sentence-transformers/README/resolve/main/sbertLogo.png" class="w-40" />
|
81 |
+
</div>
|
82 |
+
<div class="underline">Documentation</div>
|
83 |
+
</a>
|
84 |
+
<a
|
85 |
+
href="https://sbert.net/docs/package_reference/cross_encoder/cross_encoder.html#sentence_transformers.cross_encoder.CrossEncoder.push_to_hub"
|
86 |
+
class="block overflow-hidden group"
|
87 |
+
>
|
88 |
+
<div
|
89 |
+
class="w-full h-40 mb-2 bg-gray-900 group-hover:bg-gray-850 rounded-lg flex items-start justify-start overflow-hidden"
|
90 |
+
>
|
91 |
+
<img
|
92 |
+
alt=""
|
93 |
+
src="https://huggingface.co/spaces/sentence-transformers/README/resolve/main/push-to-hub.png"
|
94 |
+
class="w-full h-40 object-cover overflow-hidden"
|
95 |
+
/>
|
96 |
+
</div>
|
97 |
+
<div class="underline">Push your CrossEncoder models to the Hub ❤️ </div>
|
98 |
+
</a>
|
99 |
+
<!-- <a
|
100 |
+
href="https://huggingface.co/models?library=sentence-transformers&sort=downloads"
|
101 |
+
class="block overflow-hidden group"
|
102 |
+
>
|
103 |
+
<div
|
104 |
+
class="w-full h-40 mb-2 bg-gray-900 group-hover:bg-gray-850 rounded-lg flex items-start justify-start overflow-hidden"
|
105 |
+
>
|
106 |
+
<img
|
107 |
+
alt=""
|
108 |
+
src="https://huggingface.co/spaces/sentence-transformers/README/resolve/main/sbert-hf.png"
|
109 |
+
class="w-full h-40 object-cover overflow-hidden"
|
110 |
+
/>
|
111 |
+
</div>
|
112 |
+
<div class="underline">Find all Sentence Transformers models on the 🤗 Hub</div>
|
113 |
+
</a> -->
|
114 |
+
</div>
|
115 |
+
|
116 |
+
To upload your CrossEncoder models to the Hugging Face Hub, log in with `huggingface-cli login` and use the [`push_to_hub`](https://sbert.net/docs/package_reference/cross_encoder/cross_encoder.html#sentence_transformers.cross_encoder.CrossEncoder.push_to_hub) method within the Sentence Transformers library.
|
117 |
+
```python
|
118 |
+
from sentence_transformers import CrossEncoder
|
119 |
+
|
120 |
+
# Load or train a model
|
121 |
+
model = CrossEncoder(...)
|
122 |
+
# Push to Hub
|
123 |
+
model.push_to_hub("my_new_model")
|
124 |
+
```
|