shivi commited on
Commit
15d73ae
β€’
1 Parent(s): bb94d20

Update utils/predict.py

Browse files
Files changed (1) hide show
  1. utils/predict.py +1 -1
utils/predict.py CHANGED
@@ -9,7 +9,7 @@ import cv2
9
 
10
  #model = from_pretrained_keras("shivi/video-classification",custom_objects={"PositionalEmbedding":PositionalEmbedding,"TransformerEncoder": TransformerEncoder})
11
 
12
- model = from_pretrained_keras("shivi/video-transformers")
13
 
14
  """
15
  Below code is taken from the Video-Transformers example on keras-io by Sayak Paul
 
9
 
10
  #model = from_pretrained_keras("shivi/video-classification",custom_objects={"PositionalEmbedding":PositionalEmbedding,"TransformerEncoder": TransformerEncoder})
11
 
12
+ model = from_pretrained_keras("keras-io/video-transformers")
13
 
14
  """
15
  Below code is taken from the Video-Transformers example on keras-io by Sayak Paul