Fix CLI tool
This commit is contained in:
parent
a10569b5ab
commit
3340abbf15
|
@ -72,8 +72,8 @@ def main() -> None:
|
||||||
).serialize()
|
).serialize()
|
||||||
)
|
)
|
||||||
elif args.subparser_name == "classify":
|
elif args.subparser_name == "classify":
|
||||||
with open(args.model, "r") as f:
|
with open(args.model, "rb") as f:
|
||||||
model = json.load(f)
|
model = gptc.deserialize(f.read())
|
||||||
|
|
||||||
classifier = gptc.Classifier(model, args.max_ngram_length)
|
classifier = gptc.Classifier(model, args.max_ngram_length)
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue
Block a user