Update pipeline.py

pipeline.py

@@ -2,6 +2,7 @@ import cv2
 import numpy as np
 import tensorflow as tf
 import zipfile
+import tensorflow_addons as tfa
 
 # Set random seed for reproducibility.
 tf.random.set_seed(42)
@@ -12,8 +13,9 @@ zip_ref = zipfile.ZipFile(local_zip, 'r')
 zip_ref.extractall()
 zip_ref.close()
 
-# Load model
-model = tf.keras.models.load_model("efficientnet-b0/")
+# Load model with custom objects
+custom_objects = {"RectifiedAdam": tfa.optimizers.RectifiedAdam}
+model = tf.keras.models.load_model("efficientnet-b0/", custom_objects=custom_objects)
 
 class DetectionPipeline:
     def __init__(self, batch_size=1):