Fixed pipeline errors

pipeline.py

@@ -11,12 +11,13 @@ import numpy as np
 
 class PreTrainedPipeline():
     def __init__(self, path=""):
-        crop_size = (224, 224)
         embed_dim = 256
         num_layers = 3
         seq_length = 20
         hidden_dim = 1024
         num_heads = 8
+        self.crop_size = (224, 224)
+        self.img_size = 256
         self.nutr_names = ('energy', 'fat', 'protein', 'carbs')
         with open(os.path.join(path, "ingredients_metadata.json"), encoding='UTF-8') as f:
             self.ingredients = json.load(f)
@@ -25,7 +26,7 @@ class PreTrainedPipeline():
         self.seq_length = seq_length
 
         self.tfing = TFIng(
-            crop_size, 
+            self.crop_size, 
             embed_dim, 
             num_layers, 
             seq_length, 
@@ -38,7 +39,7 @@ class PreTrainedPipeline():
         self.tfing.load_weights(os.path.join(path, 'tfing.h5'))
 
         self.tfport = TFPort(
-            crop_size, 
+            self.crop_size, 
             embed_dim, 
             num_layers, 
             num_layers,