Rename the class

mteb_evaluate.py

@@ -1,7 +1,7 @@
 from argparse import ArgumentParser, Namespace
 from typing import List, Optional
 
-from model_api import ModelV1
+from model_api import SionicEmbeddingModel
 from mteb import MTEB
 
 RETRIEVAL_TASKS: List[str] = [
@@ -27,6 +27,7 @@ def get_arguments() -> Namespace:
     parser.add_argument('--url', type=str, default='https://api.sionic.ai/v1/embedding', help='api server url')
     parser.add_argument('--instruction', type=str, default='query: ', help='query instruction')
     parser.add_argument('--batch_size', type=int, default=128)
+    parser.add_argument('--dimension', type=int, default=2048)
     parser.add_argument('--output_dir', type=str, default='./result/v1')
     return parser.parse_args()
 
@@ -34,7 +35,7 @@ def get_arguments() -> Namespace:
 if __name__ == '__main__':
     args = get_arguments()
 
-    model = ModelV1(url=args.url, instruction=args.instruction, batch_size=args.batch_size)
+    model = SionicEmbeddingModel(url=args.url, instruction=args.instruction, batch_size=args.batch_size, dimension=args.dimension)
 
     task_names: List[str] = [t.description['name'] for t in MTEB(task_types=None, task_langs=['en']).tasks]