Update pipeline.py

pipeline.py

@@ -3,7 +3,9 @@ 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
 
+# Hugging Face expects a class named Pipeline with __call__(self, inputs)
 class Pipeline:
     def __init__(self):
         # Load tokenizer
@@ -15,9 +17,9 @@ class Pipeline:
         # Load model (SavedModel format)
         self.model = tf.keras.models.load_model(".")
 
-        # Optional: load label map
+        # Load label map if available
         self.label_map = None
-        if tf.io.gfile.exists("label_map.json"):
+        if os.path.exists("label_map.json"):
             with open("label_map.json", "r", encoding="utf-8") as f:
                 self.label_map = json.load(f)
 
@@ -30,6 +32,9 @@ class Pipeline:
         padded = pad_sequences(seq, maxlen=self.max_len, padding='post', truncating='post')
         pred_probs = self.model.predict(padded)
         pred_label = int(np.argmax(pred_probs, axis=1)[0])
+        score = float(np.max(pred_probs))
         if self.label_map:
-            return {"label": self.label_map.get(str(pred_label), pred_label), "score": float(np.max(pred_probs))}
-        return {"label": pred_label, "score": float(np.max(pred_probs))}
+            label = self.label_map.get(str(pred_label), pred_label)
+        else:
+            label = pred_label
+        return {"label": label, "score": score}