KrishnanDevilking commited on
Commit
0d82b06
·
verified ·
1 Parent(s): e363541

Upload folder using huggingface_hub

Browse files
Files changed (4) hide show
  1. README.md +10 -3
  2. inference.py +27 -0
  3. intent_model.pkl +3 -0
  4. requirements.txt +2 -0
README.md CHANGED
@@ -1,3 +1,10 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
 
1
+ # Jarvis Intent Classifier
2
+
3
+ A lightweight CPU-friendly assistant model that detects user intent and returns a response + action command.
4
+
5
+ **Architecture:** SentenceTransformer (MiniLM-L6-v2) + Logistic Regression (sklearn)
6
+ **Trained on:** custom OS assistant dataset
7
+
8
+ ### Example
9
+
10
+ Input:
inference.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pickle, random
2
+ from sentence_transformers import SentenceTransformer
3
+
4
+ # Load model once
5
+ with open("intent_model.pkl", "rb") as f:
6
+ data = pickle.load(f)
7
+
8
+ clf = data["classifier"]
9
+ id2label = data["id2label"]
10
+ embedder = SentenceTransformer(data["embed_model"])
11
+ intents_meta = data["intents_meta"]
12
+
13
+ def predict(text):
14
+ emb = embedder.encode([text])
15
+ pred = clf.predict(emb)[0]
16
+ intent = id2label[pred]
17
+ meta = intents_meta[intent]
18
+ response = random.choice(meta["responses"])
19
+ return {
20
+ "intent": intent,
21
+ "response": response,
22
+ "action": meta["action"]
23
+ }
24
+
25
+ # Required entrypoint for Hugging Face inference API
26
+ def predict_intent(inputs: str):
27
+ return predict(inputs)
intent_model.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3f89de5e99c9fb8e4a476d2827b1f577fe494d305759c139bdd66fa4e7ce1377
3
+ size 356963
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ scikit-learn
2
+ sentence-transformers