From bfe2ec8c8c4cfe97f5abfed068aabb2347b8c7b8 Mon Sep 17 00:00:00 2001
From: marcsello <punkosdmarcell@rocketmail.com>
Date: Mon, 14 Sep 2020 02:25:31 +0200
Subject: [PATCH] Fixed DB entry creation

---
 model_service/views/cnn_view.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/model_service/views/cnn_view.py b/model_service/views/cnn_view.py
index 149130e..47f61c1 100644
--- a/model_service/views/cnn_view.py
+++ b/model_service/views/cnn_view.py
@@ -47,6 +47,12 @@ class CNNView(FlaskView):
         # create bucket if necessary
         ensure_buckets()
 
+        # Create the entry in the db
+        if info['id']:
+            m = AIModel(id=info['id'], type=AIModelType.CNN)
+        else:
+            m = AIModel(type=AIModelType.CNN)
+
         # Put files into MinIO
         storage.connection.put_object(current_app.config['MINIO_CNN_BUCKET_NAME'], "model/" + str(m.id), model_file,
                                       model_file.content_length, content_type=model_file.content_type)