Ajibola commited on
Commit
b2ae985
·
1 Parent(s): f232003

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -1
README.md CHANGED
@@ -5,6 +5,7 @@ language:
5
  library_name: keras
6
  tags:
7
  - code
 
8
  ---
9
  <h1>README for Pathway Vision Transformer</h1><br>
10
 
@@ -46,4 +47,4 @@ image=cv2.resize(image, (224, 224)) #224 is the default image size
46
  image=image.image.max() #Normalize the image to [0-1]
47
  prediction=model.predict(image)
48
  prediction=np.argmax(prediction, axis=-1) #Get Highest probability class
49
- ```
 
5
  library_name: keras
6
  tags:
7
  - code
8
+ pipeline_tag: image-classification
9
  ---
10
  <h1>README for Pathway Vision Transformer</h1><br>
11
 
 
47
  image=image.image.max() #Normalize the image to [0-1]
48
  prediction=model.predict(image)
49
  prediction=np.argmax(prediction, axis=-1) #Get Highest probability class
50
+ ```