harshaUwm163 commited on
Commit
65dc354
·
1 Parent(s): e2eca4c

updated the model link to keras-io

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ import cv2
6
  # import matplotlib.pyplot as plt
7
 
8
 
9
- model = from_pretrained_keras("harsha163/learning_to_tokenize_in_ViT")
10
 
11
  # functions for inference
12
  IMG_SIZE = 32
 
6
  # import matplotlib.pyplot as plt
7
 
8
 
9
+ model = from_pretrained_keras("keras-io/learning_to_tokenize_in_ViT")
10
 
11
  # functions for inference
12
  IMG_SIZE = 32