fix model path and add config

config.json

@@ -0,0 +1,4 @@
+{
+  "model_type": "user",
+  "transformers_version": "4.20.0"
+}

modelling.py

@@ -8,7 +8,7 @@ class UserModel(PreTrainedModel):
 
     def __init__(self, config):
         super().__init__(config)
-        self.model = tf.saved_model.load('user_model/tf_retrieval_user_model')
+        self.model = tf.saved_model.load('tf_retrieval_user_model')
 
     def forward(self, user_id):
         return self.model(user_id)