Update README.md
Browse files
README.md
CHANGED
|
@@ -13,6 +13,7 @@ widget: []
|
|
| 13 |
# SentenceTransformer
|
| 14 |
|
| 15 |
This is a finetuned version of [bge-m3](https://huggingface.co/BAAI/bge-m3) for the task of SQL table retrieval.
|
|
|
|
| 16 |
|
| 17 |
## Model Details
|
| 18 |
|
|
@@ -60,9 +61,9 @@ from sentence_transformers import SentenceTransformer
|
|
| 60 |
model = SentenceTransformer("sentence_transformers_model_id")
|
| 61 |
# Run inference
|
| 62 |
sentences = [
|
| 63 |
-
'
|
| 64 |
-
|
| 65 |
-
'
|
| 66 |
]
|
| 67 |
embeddings = model.encode(sentences)
|
| 68 |
print(embeddings.shape)
|
|
|
|
| 13 |
# SentenceTransformer
|
| 14 |
|
| 15 |
This is a finetuned version of [bge-m3](https://huggingface.co/BAAI/bge-m3) for the task of SQL table retrieval.
|
| 16 |
+
<img src="https://cdn-uploads.huggingface.co/production/uploads/65cbd93c19683f98170c8d50/Y5GGGVC3RX0AUzA7oOmAe.png" width="800" height="800">
|
| 17 |
|
| 18 |
## Model Details
|
| 19 |
|
|
|
|
| 61 |
model = SentenceTransformer("sentence_transformers_model_id")
|
| 62 |
# Run inference
|
| 63 |
sentences = [
|
| 64 |
+
'What is the total biomass of fish in farms with a water temperature above 25 degrees Celsius?',
|
| 65 |
+
'CREATE TABLE Farm (FarmID INT, FarmName VARCHAR(50), WaterTemperature DECIMAL, Biomass DECIMAL)',
|
| 66 |
+
'CREATE TABLE Locations (id INT PRIMARY KEY, name VARCHAR(50), region VARCHAR(50), depth INT)',
|
| 67 |
]
|
| 68 |
embeddings = model.encode(sentences)
|
| 69 |
print(embeddings.shape)
|