Update README.md


README.md

@@ -1,5 +1,10 @@
 ---
 license: mit
+language:
+- en
+library_name: keras
+tags:
+- code
 ---
 <h1>README for Pathway Vision Transformer</h1><br>
 
@@ -42,9 +47,3 @@ 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>Prediction on Test data</h1>
-<img src='test.png' alt='evaluated predictions'>
-
-
-