1 files changed, 17 insertions, 10 deletions
diff --git a/frontend/src/app/_elements/model-load/model-load.component.ts b/frontend/src/app/_elements/model-load/model-load.component.ts
index 5f9caf9d..8bf8fd93 100644
--- a/frontend/src/app/_elements/model-load/model-load.component.ts
+++ b/frontend/src/app/_elements/model-load/model-load.component.ts
@@ -2,6 +2,7 @@ import { Component, OnInit, ViewChild, Output, EventEmitter, Input } from '@angu
 import Shared from 'src/app/Shared';
 import Experiment from 'src/app/_data/Experiment';
 import Model, { ActivationFunction, LossFunction, LossFunctionBinaryClassification, LossFunctionMultiClassification, LossFunctionRegression, Metrics, MetricsBinaryClassification, MetricsMultiClassification, MetricsRegression, NullValueOptions, Optimizer, ProblemType } from 'src/app/_data/Model';
+import { AuthService } from 'src/app/_services/auth.service';
 import { ModelsService } from 'src/app/_services/models.service';
 import { GraphComponent } from '../graph/graph.component';
 
@@ -14,7 +15,7 @@ import { GraphComponent } from '../graph/graph.component';
 export class ModelLoadComponent implements OnInit {
 
   @ViewChild(GraphComponent) graph!: GraphComponent;
-  @Input() forExperiment?:Experiment;
+  @Input() forExperiment?: Experiment;
   @Output() selectedModelChangeEvent = new EventEmitter<Model>();
 
   newModel: Model = new Model();
@@ -38,13 +39,20 @@ export class ModelLoadComponent implements OnInit {
 
   batchSizePower: number = 2;
 
-  constructor(private modelsService: ModelsService) {
+  constructor(private modelsService: ModelsService, private authService: AuthService) {
     //console.log("forExperiment = ", this.forExperiment);
-    
+    this.fetchModels();
+
+    this.authService.loggedInEvent.subscribe(_ => {
+      this.fetchModels();
+    })
+  }
+
+  fetchModels() {
     //if (this.forExperiment == undefined) {
-      this.modelsService.getMyModels().subscribe((models) => {
-        this.myModels = models;
-      });
+    this.modelsService.getMyModels().subscribe((models) => {
+      this.myModels = models;
+    });
     /*}
     else {
       this.modelsService.getMyModelsByType(ProblemType.Regression).subscribe((models) => {
@@ -56,10 +64,9 @@ export class ModelLoadComponent implements OnInit {
 
   ngOnInit(): void {
   }
-  
-  updateBatchSize()
-  {
-    this.newModel.batchSize=2**this.batchSizePower;
+
+  updateBatchSize() {
+    this.newModel.batchSize = 2 ** this.batchSizePower;
   }
 
   updateGraph() {