Update README.md

README.md

@@ -23,19 +23,29 @@ This project is licensed under the MIT License.
 
 <h1>How to use:</h1>
 <br>
-On inference</br>
+Import Librariese</br>
 
 ```ruby
-import PaViT 
+!pip install huggingface_hub["tensorflow"]
+import matplotlib.pyplot as plt
 import cv2
-from tensorflow.keras.models import *
-image=cv2.imread(image) #Load image
-image=cv2.cvtColor(image, cv2.COLOR_BGR2RGB) #Convert image to RGB
-image=cv2.resize(224, 224) #Default image size
-model=load_model('trained_weight.h5') #Load weight
-prediction=model.predict(image) #run inference
-prediction=np.argmax(prediction, axis=-1) #Show highest probability class
+from huggingface_hub import from_pretrained_keras
 ```
+<h1>On inference</h1><br>
+```ruby 
+#load model
+model=from_pretrained_keras('Ajibola/PaViT')
+
+#load image
+image=cv2.imread('image_path')
+image=cv2.resize(image, (224, 224)) #224 is the default image size
+image=image.image.max() #Normalize the image to [0-1]
+prediction=model.predict(image)
+prediction=np.argmax(prediction, axis=-1) #Get Highest probability class
+
+
+```
+
 <br>On Training</br>
 ```ruby
 model=PaViT.PaViT()