1 files changed, 4 insertions, 0 deletions
diff --git a/frontend/src/app/_pages/add-model/add-model.component.ts b/frontend/src/app/_pages/add-model/add-model.component.ts
index 3c8c32c6..c18ad324 100644
--- a/frontend/src/app/_pages/add-model/add-model.component.ts
+++ b/frontend/src/app/_pages/add-model/add-model.component.ts
@@ -13,6 +13,7 @@ import { ModelsService } from 'src/app/_services/models.service';
 export class AddModelComponent implements OnInit {
 
   @ViewChild(DatasetLoadComponent) datasetLoadComponent?: DatasetLoadComponent;
+  datasetLoaded: boolean = false;
 
   newModel: Model;
 
@@ -31,6 +32,9 @@ export class AddModelComponent implements OnInit {
   }
 
   addModel() {
+    if (this.datasetLoadComponent)
+      this.models.addDataset(this.datasetLoadComponent?.dataset);
+
     this.getCheckedInputCols();
     this.getCheckedOutputCol();
     if (this.validationInputsOutput())