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)