Upload 5 files
Browse files- handler.py +1 -6
- logistic_classifier.joblib +1 -1
- tokenizer.json +0 -0
- vectorizer.joblib +2 -2
handler.py
CHANGED
|
@@ -8,12 +8,6 @@ class EndpointHandler:
|
|
| 8 |
self.vectorizer = joblib.load(os.path.join(model_dir, 'vectorizer.joblib'))
|
| 9 |
self.model = joblib.load(os.path.join(model_dir, 'logistic_classifier.joblib'))
|
| 10 |
|
| 11 |
-
# Define the custom tokenizer function
|
| 12 |
-
def custom_tokenizer(text):
|
| 13 |
-
return text.split()
|
| 14 |
-
|
| 15 |
-
self.tokenizer = custom_tokenizer
|
| 16 |
-
|
| 17 |
# Verify that the tokenizer configuration is correct
|
| 18 |
with open(os.path.join(model_dir, "tokenizer.json"), "r") as file:
|
| 19 |
tokenizer_config = json.load(file)
|
|
@@ -53,3 +47,4 @@ class EndpointHandler:
|
|
| 53 |
return json.dumps({"error": str(e)})
|
| 54 |
|
| 55 |
|
|
|
|
|
|
| 8 |
self.vectorizer = joblib.load(os.path.join(model_dir, 'vectorizer.joblib'))
|
| 9 |
self.model = joblib.load(os.path.join(model_dir, 'logistic_classifier.joblib'))
|
| 10 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
# Verify that the tokenizer configuration is correct
|
| 12 |
with open(os.path.join(model_dir, "tokenizer.json"), "r") as file:
|
| 13 |
tokenizer_config = json.load(file)
|
|
|
|
| 47 |
return json.dumps({"error": str(e)})
|
| 48 |
|
| 49 |
|
| 50 |
+
handler = EndpointHandler("Dineflow/Restaurant-Feedback-Analysis-Model/tree/main")
|
logistic_classifier.joblib
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 200927
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:62378fbef899119b4cecc05d3a98e1c1f1610657959cb90aa89b672a6d49a99d
|
| 3 |
size 200927
|
tokenizer.json
CHANGED
|
The diff for this file is too large to render.
See raw diff
|
|
|
vectorizer.joblib
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
-
size
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:30d163d2ed7250473510b1a778110399804150eaf11de344d794a6f085b972fc
|
| 3 |
+
size 162168
|