Update README.md

README.md

@@ -40,20 +40,8 @@ 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
-
 ```
-<h1>On Training</h1><br>
-```ruby
-model=PaViT.PaViT() 
-#Output means the unit default is 15 and activation 'sigmoid'
-#Output_class default is None so it uses Dense layer as the output_layer
-#Ouptut_class is the output layer 
-p_model=model.model(output_class=None, activation='softmax', output=15) 
-p_model.summary()
-p_model.compile(...)
-p_model.fit(...)
 
-```
 <h1>Prediction on Test data</h1>
 <img src='https://huggingface.co/Ajibola/PaViT/blob/main/test.png' alt='evaluated predictions'>