Fix issues from @angyaloliver's feedback
This commit is contained in:
parent
584d746447
commit
041eefc401
12 changed files with 30 additions and 26 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue