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)
|