Model save

README.md

@@ -1,5 +1,5 @@
 ---
-pipeline_tag: feature-extraction
+license: apache-2.0
 tags:
 - generated_from_trainer
 model-index:
@@ -12,7 +12,7 @@ should probably proofread and complete it, then remove this comment. -->
 
 # bert-geolocation-prediction
 
-This model is a fine-tuned version of [](https://huggingface.co/) on an unknown dataset.
+This model is a fine-tuned version of [bert-base-multilingual-cased](https://huggingface.co/bert-base-multilingual-cased) on an unknown dataset.
 
 ## Model description
 

config.json

@@ -11,7 +11,6 @@
   "intermediate_size": 3072,
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
-  "model_type": "bert-geo-regressor",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 0,

handler.py

@@ -0,0 +1,27 @@
+from typing import Dict, List, Any
+import torch 
+from transformers import pipeline
+
+class EndpointHandler():
+    def __init__(self, path=""):
+        self.pipeline = pipeline("geo-regression", model=path)
+
+    def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
+        """
+         data args:
+              inputs (:obj: `str`)
+              date (:obj: `str`)
+        Return:
+              A :obj:`list` | `dict`: will be serialized and returned
+        """
+        # get inputs
+        inputs = data.pop("inputs", data)
+        # get additional date field
+        date = data.pop("date", None)
+
+        # check if date exists and if it is a holiday
+
+        # run normal prediction
+        prediction = self.pipeline(inputs)
+        return prediction
+

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7624d9e456ab13265c8c0860c51ae9c48300554b0eb3c1bc46875abed2f3a8ee
+oid sha256:c09a0aba06c9e94944778dfd371d7dec4d9bcf6dcf60ab2481bbda48ecef0e34
 size 711543661

requirments.txt

@@ -0,0 +1,2 @@
+pipe
+handler

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:aab6fdd49d925c2befeaa9bcc47e8b66b1120c29354f0e9fe5dc32701773e737
+oid sha256:5446b669b3602b87aff7a66261257635a4e1337e7d796c9fa0eacad20dfd36e1
 size 3439