Simple debug version

pipeline.py

@@ -11,15 +11,15 @@ from PIL import Image
 # most of this code has been obtained from Datature's prediction script
 # https://github.com/datature/resources/blob/main/scripts/bounding_box/prediction.py
 
-def load_model():
-	return tf.saved_model.load('./saved_model')
+# def load_model():
+# 	return tf.saved_model.load('./saved_model')
 
-model = load_model()
+# model = load_model()
 
-# class PreTrainedPipeline():
-#     def __init__(self, path: str):
-#         # load the model
-#         self.model = tf.saved_model.load(os.path.join(path, "saved_model"))
+class PreTrainedPipeline():
+    def __init__(self, path: str):
+        # load the model
+        self.model = tf.saved_model.load(os.path.join(path, "saved_model"))
 
 #     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]:
 
@@ -81,12 +81,12 @@ model = load_model()
         #         "score": 1.0,
         #     })
 		
-labels = [{"score":0.9509243965148926,"label":"car","box":{"xmin":142,"ymin":106,"xmax":376,"ymax":229}},
-			{"score":0.9981777667999268,"label":"car","box":{"xmin":405,"ymin":146,"xmax":640,"ymax":297}},
-			{"score":0.9963648915290833,"label":"car","box":{"xmin":0,"ymin":115,"xmax":61,"ymax":167}},
-			{"score":0.974663257598877,"label":"car","box":{"xmin":155,"ymin":104,"xmax":290,"ymax":141}},
-			{"score":0.9986898303031921,"label":"car","box":{"xmin":39,"ymin":117,"xmax":169,"ymax":188}},
-			{"score":0.9998276233673096,"label":"person","box":{"xmin":172,"ymin":60,"xmax":482,"ymax":396}},
-			{"score":0.9996274709701538,"label":"skateboard","box":{"xmin":265,"ymin":348,"xmax":440,"ymax":413}}]
+# labels = [{"score":0.9509243965148926,"label":"car","box":{"xmin":142,"ymin":106,"xmax":376,"ymax":229}},
+# 			{"score":0.9981777667999268,"label":"car","box":{"xmin":405,"ymin":146,"xmax":640,"ymax":297}},
+# 			{"score":0.9963648915290833,"label":"car","box":{"xmin":0,"ymin":115,"xmax":61,"ymax":167}},
+# 			{"score":0.974663257598877,"label":"car","box":{"xmin":155,"ymin":104,"xmax":290,"ymax":141}},
+# 			{"score":0.9986898303031921,"label":"car","box":{"xmin":39,"ymin":117,"xmax":169,"ymax":188}},
+# 			{"score":0.9998276233673096,"label":"person","box":{"xmin":172,"ymin":60,"xmax":482,"ymax":396}},
+# 			{"score":0.9996274709701538,"label":"skateboard","box":{"xmin":265,"ymin":348,"xmax":440,"ymax":413}}]
 
 # return labels