kaixkhazaki
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -59,4 +59,12 @@ Test Loss: 0.8134, Test Acc: 81.56%
|
|
59 |
|
60 |
|
61 |
## Usage
|
|
|
|
|
62 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
59 |
|
60 |
|
61 |
## Usage
|
62 |
+
```python
|
63 |
+
from transformers import pipeline
|
64 |
|
65 |
+
# Load the model using the image-classification pipeline
|
66 |
+
pipe = pipeline("image-classification", model="kaixkhazaki/vit_doclaynet_base")
|
67 |
+
|
68 |
+
# Test it with an image
|
69 |
+
result = pipe("path_to_image.jpg")
|
70 |
+
print(result)
|