Fix bug.

README.md

@@ -13,14 +13,14 @@ pip install -r requirements.txt
 ## usage
 
 ```
-usage: run.py [-h] (--dir DIR | --file FILE) [--threthold THRETHOLD] [--ext EXT]
+usage: run.py [-h] (--dir DIR | --file FILE) [--threshold THRETHOLD] [--ext EXT]
               [--model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}]
 
 options:
   -h, --help            show this help message and exit
   --dir DIR             ディレクトリ内の画像すべてに予測を行う
   --file FILE           ファイルに対して予測を行う
-  --threthold THRETHOLD
+  --threshold THRETHOLD
                         予測値の足切り確率（デフォルトは0.35）
   --ext EXT             dirの場合にキャプションファイルにつける拡張子
   --model {wd14-vit.v1,wd14-vit.v2,wd14-convnext.v1,wd14-convnext.v2,wd14-convnextv2.v1,wd14-swinv2-v1,wd-v1-4-moat-tagger.v2,mld-caformer.dec-5-97527,mld-tresnetd.6-30000}

run.py

@@ -15,7 +15,7 @@ group.add_argument('--dir', help='ディレクトリ内の画像すべてに予
 group.add_argument('--file', help='ファイルに対して予測を行う')
 
 parser.add_argument(
-    '--threthold',
+    '--threshold',
     type=float,
     default=0.35,
     help='予測値の足切り確率（デフォルトは0.35）')
@@ -40,7 +40,7 @@ def image_interrogate(image_path: Path):
     """
     im = Image.open(image_path)
     result = interrogator.interrogate(im)
-    return Interrogator.postprocess_tags(result[1], threshold=0.35)
+    return Interrogator.postprocess_tags(result[1], threshold=args.threshold)
 
 if args.dir:
     d = Path(args.dir)