Update pipeline.py

pipeline.py

@@ -3,18 +3,18 @@ from typing import Any, Dict, List
 
 import tensorflow as tf
 from tensorflow import keras
-from huggingface_hub import from_pretrained_keras, hf_hub_download
 import base64
 import io
+import os
 import numpy as np
 from PIL import Image
 
 
 
 class PreTrainedPipeline():
-    def __init__(self, model_id: str):
+    def __init__(self, path: str):
 
-        self.model = keras.models.load_model("./tf_model.h5")
+        self.model = keras.models.load_model(os.path.join(path, "tf_model.h5"))
 
     def __call__(self, inputs: "Image.Image")-> List[Dict[str, Any]]: