kaixkhazaki commited on
Commit
bb37265
·
verified ·
1 Parent(s): 8fe318c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +8 -0
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)