feat set kimchima_cli back

Signed-off-by: micost <chuan.z@hotmail.com>
SkywardAI · May 22, 2024 · 8faa489 · 8faa489
1 parent a459b16
commit 8faa489
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/kimchima/cmds/kimchima_cli.py b/src/kimchima/cmds/kimchima_cli.py
@@ -14,7 +14,7 @@
 
 import argparse
 
-from kimchima.cmds.auto_cli import CommandCrossEncoder
+from kimchima.cmds.auto_cli import CommandAutoModel
 
 
 def main():
@@ -31,7 +31,7 @@ def main():
     parser_auto=subparsers.add_parser("auto", help="auto help")
     parser_auto.add_argument("model_name_or_path", default="sentence-transformers/all-MiniLM-L6-v2", help="model name or path")
     parser_auto.add_argument("text", help="text str or list of text str")
-    parser_auto.set_defaults(func=CommandCrossEncoder.auto)
+    parser_auto.set_defaults(func=CommandAutoModel.auto)
 
     args = parser.parse_args()
     args.func(args)