Update utils/predict.py

utils/predict.py

@@ -1,4 +1,4 @@
-from .custom_layers import TransformerEncoder, PositionalEmbedding
+#from .custom_layers import TransformerEncoder, PositionalEmbedding
 from .constants import MAX_SEQ_LENGTH, NUM_FEATURES, IMG_SIZE, CLASS_VOCAB
 from huggingface_hub import from_pretrained_keras
 from tensorflow import keras
@@ -7,10 +7,13 @@ import numpy as np
 import imageio
 import cv2
 
-model = from_pretrained_keras("shivi/video-classification",custom_objects={"PositionalEmbedding":PositionalEmbedding,"TransformerEncoder": TransformerEncoder})
+#model = from_pretrained_keras("shivi/video-classification",custom_objects={"PositionalEmbedding":PositionalEmbedding,"TransformerEncoder": TransformerEncoder})
 
-# model = from_pretrained_keras("shivi/video-transformers")
+model = from_pretrained_keras("shivi/video-transformers")
 
+"""
+Below code is taken from the Video-Transformers on keras-io example by Sayak Paul
+"""
 def build_feature_extractor():
     feature_extractor = keras.applications.DenseNet121(
         weights="imagenet",