diff --git a/gptc/__main__.py b/gptc/__main__.py index 26769cb..d98d4bd 100644 --- a/gptc/__main__.py +++ b/gptc/__main__.py @@ -72,8 +72,8 @@ def main() -> None: ).serialize() ) elif args.subparser_name == "classify": - with open(args.model, "r") as f: - model = json.load(f) + with open(args.model, "rb") as f: + model = gptc.deserialize(f.read()) classifier = gptc.Classifier(model, args.max_ngram_length)