vkuropiatnyk commited on
Commit
17752aa
·
verified ·
1 Parent(s): e3a3538

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -38,8 +38,8 @@ This is an image classification model based on **Google EfficientNet-B0**, fine-
38
 
39
 
40
 
41
- ### How to use
42
- Example of how to classify an image into one of the 39 classes using transformers:
43
 
44
  ```python
45
  import torch
@@ -96,7 +96,8 @@ for idx, probs_image in enumerate(probs_batch):
96
  ```
97
 
98
 
99
- Example of how to classify an image into one of the 39 classes using onnx runtime:
 
100
 
101
  ```python
102
  import onnxruntime
 
38
 
39
 
40
 
41
+ ### How to use - Transformers
42
+ Example of how to classify an image into one of the 26 classes using transformers:
43
 
44
  ```python
45
  import torch
 
96
  ```
97
 
98
 
99
+ ### How to use - ONNX
100
+ Example of how to classify an image into one of the 26 classes using onnx runtime:
101
 
102
  ```python
103
  import onnxruntime