diff --git a/main.py b/main.py index 690859b..103acdf 100644 --- a/main.py +++ b/main.py @@ -17,7 +17,7 @@ import plotly.io as pio from collections import defaultdict, Counter from pathlib import Path from urllib.parse import urlparse -from typing import Any, Optional, List, Set, Tuple +from typing import Any, Optional, List, Set, Tuple, Dict from aiolimiter import AsyncLimiter from dataclasses import dataclass, field @@ -31,7 +31,7 @@ DEFAULTS = [ "wd14-convnext.v1", "wd14-convnext.v2", "wd14-convnextv2.v1", - "wd14-swinv2-v1", + # "wd14-swinv2-v1", "wd-v1-4-moat-tagger.v2", "mld-caformer.dec-5-97527", # broken model: "mld-tresnetd.6-30000",