Skip to content

Commit

Permalink
make it better
Browse files Browse the repository at this point in the history
  • Loading branch information
davidberenstein1957 committed Nov 21, 2022
1 parent 8e939c1 commit beb2eaa
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
7 changes: 6 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -13,7 +13,12 @@ or install with faster inference using onnx.

``` pip install classy-classification[onnx]```

## ONNX on Mac M1
## ONNX issues

### pickling

ONNX does show some issues when pickling the data.
### M1
Some [installation issues](https://github.com/onnx/onnx/issues/3129) might occur, which can be fixed by these commands.

```
Expand Down
1 change: 0 additions & 1 deletion test.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,4 +7,3 @@
"text_categorizer",
config={"data": list(training_data), "cat_type": "zero", "include_sent": True, "multi_label": True},
)
print(nlp("kitchen stuff")._.cats)

0 comments on commit beb2eaa

Please sign in to comment.