diff --git a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-embed-ui/machine-learning-embed-ui.component.html b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-embed-ui/machine-learning-embed-ui.component.html
index 39f557effb..01df45c3dc 100644
--- a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-embed-ui/machine-learning-embed-ui.component.html
+++ b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-embed-ui/machine-learning-embed-ui.component.html
@@ -198,7 +198,7 @@ <h3 mat-dialog-title class="mb-3">Embed Chatbot or AI Search on your website</h3
   </a>
 
   <button
-    mat-flat-button
+    mat-button
     (click)="embed()">
     Copy
   </button>
diff --git a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import-feedback/machine-learning-import-feedback.component.html b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import-feedback/machine-learning-import-feedback.component.html
index 798aec73bb..7b07599466 100644
--- a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import-feedback/machine-learning-import-feedback.component.html
+++ b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import-feedback/machine-learning-import-feedback.component.html
@@ -21,7 +21,8 @@ <h3 mat-dialog-title class="mb-0">Progress</h3>
 <div mat-dialog-actions [align]="'end'">
 
   <button
-    mat-button
+    mat-flat-button
+    color="primary"
     mat-dialog-close>
     Close
   </button>
diff --git a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.html b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.html
index f52e4405e8..2a8656d905 100644
--- a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.html
+++ b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.html
@@ -117,6 +117,20 @@ <h3 mat-dialog-title class="mb-0">Import <span class="d-md-inline d-none">traini
         </div>
 
       </form>
+
+      <div class="text-end me-2">
+
+        <mat-slide-toggle
+          labelPosition="before"
+          class="me-2"
+          matTooltip="Summarize training snippets that are too long to be effectively used"
+          [disabled]="uploading"
+          [(ngModel)]="summarize">
+          Summarize
+        </mat-slide-toggle>
+
+      </div>
+
     </mat-tab>
 
     <mat-tab>
@@ -317,20 +331,12 @@ <h4 class="mb-0 mt-3 text-center">{{getFileName()}}</h4>
 
 <div mat-dialog-actions [align]="'end'">
 
-  <mat-slide-toggle
-    labelPosition="before"
-    class="me-2"
-    matTooltip="Summarize training snippets that are too long to be effectively used"
-    [disabled]="uploading"
-    [(ngModel)]="summarize">
-    Summarize
-  </mat-slide-toggle>
-
   <button
-    mat-button
+    mat-flat-button
+    color="primary"
     [disabled]="uploading"
     mat-dialog-close>
-    Cancel
+    Close
   </button>
 
 </div>
diff --git a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.ts b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.ts
index 215d7c2c8d..e0967641d3 100644
--- a/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.ts
+++ b/frontend/src/app/components/protected/manage/machine-learning/components/machine-learning-import/machine-learning-import.component.ts
@@ -79,6 +79,7 @@ export class MachineLearningImportComponent {
     this.uploadIndex = 0;
     this.uploadCount = 0;
     this.files = event.target.files;
+    this.generalService.showLoading();
     this.uploadCurrentFile();
   }
 
@@ -115,6 +116,7 @@ export class MachineLearningImportComponent {
           // Incrementing upload index
           this.uploadIndex += 1;
           if (this.uploadIndex >= this.files.length) {
+            this.generalService.hideLoading();
             this.generalService.showFeedback(`${this.uploadCount} training snippets successfully imported`, 'successMessage');
             this.uploading = false;
             this.trainingFileModel = '';