Satyam0077 commited on
Commit
4c3c772
·
verified ·
1 Parent(s): cc5ebd2

Delete inference.py

Browse files
Files changed (1) hide show
  1. inference.py +0 -39
inference.py DELETED
@@ -1,39 +0,0 @@
1
- import os
2
- from src.preprocessing import clean_text
3
- from src.features import create_features
4
- from src.model import load_model
5
- from src.entity_extraction import extract_entities
6
- import numpy as np
7
- import joblib
8
- import scipy.sparse
9
- from textblob import TextBlob
10
-
11
- # ✅ Correct path to model files
12
- BASE_PATH = os.path.join(os.path.dirname(__file__), 'models')
13
-
14
- # ✅ Load models from correct location
15
- model_issue = load_model(os.path.join(BASE_PATH, 'model_issue_type.pkl'))
16
- model_urgency = load_model(os.path.join(BASE_PATH, 'model_urgency_level.pkl'))
17
- tfidf = joblib.load(os.path.join(BASE_PATH, 'tfidf.pkl'))
18
-
19
- def predict_ticket(ticket_text):
20
- clean = clean_text(ticket_text)
21
- X_tfidf = tfidf.transform([clean])
22
- ticket_length = len(clean.split())
23
- sentiment = TextBlob(clean).sentiment.polarity
24
-
25
- X_features = scipy.sparse.hstack([
26
- X_tfidf,
27
- np.array([[ticket_length]]),
28
- np.array([[sentiment]])
29
- ])
30
-
31
- issue_pred = model_issue.predict(X_features)[0]
32
- urgency_pred = model_urgency.predict(X_features)[0]
33
- entities = extract_entities(ticket_text)
34
-
35
- return {
36
- 'issue_type': issue_pred,
37
- 'urgency_level': urgency_pred,
38
- 'entities': entities
39
- }