Update inference.py

inference.py

@@ -10,7 +10,7 @@ NAME_TO_PC = {
 }
 NOTE_TOKEN_RE = re.compile(r"[A-Ga-g](?:#|b)?")
 
-# Load model at startup
+# Load model
 MODEL_PATH = os.path.join(os.path.dirname(__file__), "chord_classifier.pkl")
 clf = joblib.load(MODEL_PATH)
 
@@ -23,12 +23,13 @@ def notes_to_vector(notes_str: str):
         vec[p] = 1
     return vec
 
-def predict(notes: str):
+# 👇 Hugging Face Inference API entrypoint
+def predict(inputs: str):
     """
-    Main inference function.
     Hugging Face will call this with {"inputs": "C E G"}.
+    Must return JSON-serializable output.
     """
-    vec = notes_to_vector(notes)
+    vec = notes_to_vector(inputs)
     if np.sum(vec) < 2:
         return {"label": "Invalid input (need 2+ notes)"}
     label = clf.predict([vec])[0]