Fixed error in pipeline

pipeline.py

@@ -8,11 +8,12 @@ import tensorflow as tf
 
 class PreTrainedPipeline():
     def __init__(self, path=""):
-        crop_size = (224, 224)
+        self.crop_size = (224, 224)
         self.nutr_names = ('energy', 'fat', 'protein', 'carbs')
-        self.model = FCNutr(self.nutr_names, crop_size, 4096, 3, False)
+        self.img_size = 256
+        self.model = FCNutr(self.nutr_names, self.crop_size, 4096, 3, False)
         self.model.compile()
-        self.model(tf.zeros((1, crop_size[0], crop_size[1], 3)))
+        self.model(tf.zeros((1, self.crop_size[0], self.crop_size[1], 3)))
         self.model.load_weights(os.path.join(path, "fcnutr.h5"))
 
     def __call__(self, inputs: "Image.Image") -> List[Dict[str, Any]]: