Upload pipeline_full.joblib + model.joblib + inference.py

README.md

@@ -1 +1,6 @@
-# Model for cosmetic-category-model\n\nThis folder contains:\n- model.joblib\n- inference.py\n- requirements.txt\n\nInference expects JSON like {"text":"raw product text"} or {"inputs":{"features":[...]}}
+# pipeline upload
+This repo contains the saved pipeline and model for the cosmetic category model.
+Files:
+- pipeline_full.joblib
+- model.joblib
+- inference.py

inference.py

@@ -1,21 +1,21 @@
-# inference.py (auto-generated) - separate vectorizer support
-import joblib, numpy as np
-MODEL = joblib.load("model.joblib")
-VECT = joblib.load("vect_f.joblib")\n
+import joblib, numpy as np, os
+MODEL = joblib.load("pipeline_full.joblib") if os.path.exists("pipeline_full.joblib") else joblib.load("model.joblib")
+
 def run(inputs):
     try:
+        # expecting either {"text":"..."} or raw list
         if isinstance(inputs, dict):
             if "text" in inputs:
                 X = [inputs["text"]]
-            elif "features" in inputs:
-                X = [inputs["features"]]
+            elif "inputs" in inputs and isinstance(inputs["inputs"], dict) and "features" in inputs["inputs"]:
+                X = [inputs["inputs"]["features"]]
             else:
                 X = [inputs]
         else:
             X = [inputs]
-        # apply TF-IDF vectorizer to incoming text
-        if isinstance(X[0], str):\n            X_trans = VECT.transform(X)\n        else:\n            # assume already numeric features\n            X_trans = X\n
-        pred = MODEL.predict_proba(X_trans).tolist() if hasattr(MODEL, 'predict_proba') else MODEL.predict(X_trans).tolist()
-        return { "prediction": pred }
+        if hasattr(MODEL, "predict_proba"):
+            return {"prediction": MODEL.predict_proba(X).tolist()}
+        else:
+            return {"prediction": MODEL.predict(X).tolist()}
     except Exception as e:
-        return { "error": str(e) }
+        return {"error": str(e)}

model.joblib

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:01fa97214e2006305b631baeb971a49a69364da975da0d072c05bc77fdb3d72c
-size 1784996
+oid sha256:3f2f37b0995b3d162b1bd9d7444ae2c49c8ccfae26fc030b75845a597749e7d9
+size 4169433

pipeline_full.joblib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f2f37b0995b3d162b1bd9d7444ae2c49c8ccfae26fc030b75845a597749e7d9
+size 4169433