Update README.md
Browse files
README.md
CHANGED
@@ -31,6 +31,7 @@ import cv2
|
|
31 |
from huggingface_hub import from_pretrained_keras
|
32 |
```
|
33 |
<h1>On inference</h1><br>
|
|
|
34 |
```ruby
|
35 |
#load model
|
36 |
model=from_pretrained_keras('Ajibola/PaViT')
|
@@ -43,7 +44,7 @@ prediction=np.argmax(prediction, axis=-1) #Get Highest probability class
|
|
43 |
```
|
44 |
|
45 |
<h1>Prediction on Test data</h1>
|
46 |
-
<img src='
|
47 |
|
48 |
|
49 |
|
|
|
31 |
from huggingface_hub import from_pretrained_keras
|
32 |
```
|
33 |
<h1>On inference</h1><br>
|
34 |
+
|
35 |
```ruby
|
36 |
#load model
|
37 |
model=from_pretrained_keras('Ajibola/PaViT')
|
|
|
44 |
```
|
45 |
|
46 |
<h1>Prediction on Test data</h1>
|
47 |
+
<img src='test.png' alt='evaluated predictions'>
|
48 |
|
49 |
|
50 |
|