Fix saving of namecat model
diff --git a/tf-ner-poc/src/main/python/namecat/namecat.py b/tf-ner-poc/src/main/python/namecat/namecat.py
index 21c56ff..2a6e23f 100644
--- a/tf-ner-poc/src/main/python/namecat/namecat.py
+++ b/tf-ner-poc/src/main/python/namecat/namecat.py
@@ -208,7 +208,7 @@
with TemporaryDirectory() as temp_dir:
temp_model_dir = temp_dir + "/model"
- builder = tf.saved_model.builder.SavedModelBuilder("./namecat_model" + str(epoch))
+ builder = tf.saved_model.builder.SavedModelBuilder(temp_model_dir)
builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING])
builder.save()
