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