From b533dd041d2ad5db1c568bafa4a645cf31101afe Mon Sep 17 00:00:00 2001
From: Nicolas Mowen <nickmowen213@gmail.com>
Date: Wed, 5 Nov 2025 05:09:21 -0700
Subject: [PATCH] Cleanup desktop

---
 web/src/views/classification/ModelTrainingView.tsx | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/web/src/views/classification/ModelTrainingView.tsx b/web/src/views/classification/ModelTrainingView.tsx
index 2eecf48f9..a27a06a9e 100644
--- a/web/src/views/classification/ModelTrainingView.tsx
+++ b/web/src/views/classification/ModelTrainingView.tsx
@@ -327,7 +327,7 @@ export default function ModelTrainingView({ model }: ModelTrainingViewProps) {
       </AlertDialog>
 
       <div className="flex flex-row justify-between gap-2 p-2 align-middle">
-        {!selectedImages?.length && (
+        {(isDesktop || !selectedImages?.length) && (
           <div className="flex flex-row items-center justify-center gap-2">
             <Button
               className="flex items-center gap-2.5 rounded-lg"
@@ -353,8 +353,13 @@ export default function ModelTrainingView({ model }: ModelTrainingViewProps) {
           </div>
         )}
         {selectedImages?.length > 0 ? (
-          <div className="flex w-full items-center justify-between gap-2">
-            <div className="mx-1 flex w-48 items-center justify-center text-sm text-muted-foreground">
+          <div
+            className={cn(
+              "flex w-full items-center justify-end gap-2",
+              isMobileOnly && "justify-between",
+            )}
+          >
+            <div className="flex w-48 items-center justify-center text-sm text-muted-foreground">
               <div className="p-1">{`${selectedImages.length} selected`}</div>
               <div className="p-1">{"|"}</div>
               <div