Fix issues from @angyaloliver's feedback

This commit is contained in:
Andras Schmelczer 2022-07-13 20:23:56 +02:00
parent 584d746447
commit 041eefc401
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
12 changed files with 30 additions and 26 deletions

View file

@ -62,7 +62,7 @@
"\n",
"@GreatAI.create\n",
"@use_model(\"small-domain-prediction\", version=\"latest\")\n",
"@parameter(\"target_confidence\", validator=lambda c: 0 <= c <= 100)\n",
"@parameter(\"target_confidence\", validate=lambda c: 0 <= c <= 100)\n",
"def predict_domain(\n",
" text: str, model: Pipeline, target_confidence: int = 50\n",
") -> MultiLabelClassificationOutput:\n",