Update README

neuralmagic · Oct 18, 2023 · 636f942 · 636f942
1 parent 05cdb9c
commit 636f942
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/deepsparse/sentence_transformers/README.md b/src/deepsparse/sentence_transformers/README.md
@@ -57,8 +57,8 @@ from mteb import MTEB
 model_name = "TaylorAI/bge-micro-v2"
 
 # DeepSparse Model Evaluation
-import deepsparse
-model = deepsparse.sentence_transformers.SentenceTransformer(model_name, export=True)
+from deepsparse.sentence_transformers import SentenceTransformer
+model = SentenceTransformer(model_name, export=True)
 evaluation = MTEB(tasks=["Banking77Classification"])
 results_ds = evaluation.run(model, output_folder=f"results/ds-{model_name}")
 print(results_ds)
@@ -83,4 +83,4 @@ This script performs a comparative analysis between the DeepSparse-optimized mod
 
 This documentation is based on the original README from [SentenceTransformers](https://www.sbert.net/). It extends the original functionalities with the optimizations provided by [DeepSparse](https://github.com/neuralmagic/deepsparse).
 
-**Note**: The example usage is designed for the DeepSparse-enhanced version of SentenceTransformers. Make sure to follow the specific installation instructions for full compatibility. Performance optimizations with batching and other advanced features will be part of future updates.
+**Note**: The example usage is designed for the DeepSparse-enhanced version of SentenceTransformers. Make sure to follow the specific installation instructions for full compatibility. Performance optimizations with batching and other advanced features will be part of future updates.