Update pipeline.py

pipeline.py

@@ -3,7 +3,6 @@ import tensorflow as tf
 from tensorflow.keras.preprocessing.sequence import pad_sequences
 from tensorflow.keras.preprocessing.text import tokenizer_from_json
 import json
-import os
 
 class Pipeline:
     def __init__(self):
@@ -16,9 +15,9 @@ class Pipeline:
         # Load model (SavedModel format)
         self.model = tf.keras.models.load_model(".")
 
-        # Optionally, load label map if you have one
+        # Optional: load label map
         self.label_map = None
-        if os.path.exists("label_map.json"):
+        if tf.io.gfile.exists("label_map.json"):
             with open("label_map.json", "r", encoding="utf-8") as f:
                 self.label_map = json.load(f)