Fixed DB entry creation

2020-09-14 02:25:31 +02:00 · 2020-09-14 02:25:31 +02:00 · bfe2ec8c8c
parent bf18d8e74f
commit bfe2ec8c8c
1 changed files with 6 additions and 0 deletions
--- 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)