Upload 2 files

model.pkl

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

model_wrapper.py

@@ -0,0 +1,17 @@
+# model_wrapper.py
+
+import joblib
+import numpy as np
+
+LABELS = ['admiration','anger','disgust','fear','hope',
+          'joy','love','pride','sadness']
+
+class MyModel:
+    def __init__(self):
+        bundle = joblib.load("model.pkl")
+        self.vectorizer = bundle["vectorizer"]
+        self.classifier = bundle["classifier"]
+
+    def predict(self, texts):
+        X = self.vectorizer.transform(texts)
+        return self.classifier.predict(X)