Update pipeline.py

pipeline.py

@@ -7,9 +7,9 @@ import pickle
 from PIL import Image
 
 class PreTrainedPipeline():
-    def __init__(self, path="model"):
+    def __init__(self, path: str):
         
-        self.model = keras.models.load_model(os.path.join(path, ""))
+        self.model = keras.models.load_model(os.path.join(path, "model"))
 
         self.word_to_index = tf.keras.layers.StringLookup(
           mask_token="",