Create inference.py

inference.py

@@ -0,0 +1,22 @@
+from env import EmailEnv
+
+env = EmailEnv()
+
+def predict():
+    email = env.reset()
+    text = email["text"]
+
+    if "meeting" in text.lower():
+        action = {"label": "important", "reply": "Sure, I will attend the meeting."}
+    elif "lottery" in text.lower():
+        action = {"label": "spam", "reply": "This looks like spam."}
+    else:
+        action = {"label": "normal", "reply": "Sounds good!"}
+
+    reward = env.step(action)
+
+    return {
+        "email": text,
+        "action": action,
+        "reward": reward
+    }