Sync from GitHub Actions
Browse files
README.md
CHANGED
|
@@ -52,8 +52,8 @@ sys.path.insert(0, model_dir)
|
|
| 52 |
|
| 53 |
from src.inference import EmbeddingInference
|
| 54 |
|
| 55 |
-
# Load
|
| 56 |
-
model = EmbeddingInference.from_pretrained(
|
| 57 |
|
| 58 |
# 1. Similarity
|
| 59 |
score = model.similarity("Machine learning is great", "AI is wonderful")
|
|
|
|
| 52 |
|
| 53 |
from src.inference import EmbeddingInference
|
| 54 |
|
| 55 |
+
# Load model
|
| 56 |
+
model = EmbeddingInference.from_pretrained(model_dir)
|
| 57 |
|
| 58 |
# 1. Similarity
|
| 59 |
score = model.similarity("Machine learning is great", "AI is wonderful")
|