diff --git a/apps/computer-vision/app/text_to_image/index.tsx b/apps/computer-vision/app/text_to_image/index.tsx
index eff457dc5..266cbed44 100644
--- a/apps/computer-vision/app/text_to_image/index.tsx
+++ b/apps/computer-vision/app/text_to_image/index.tsx
@@ -18,7 +18,7 @@ export default function TextToImageScreen() {
   const [inferenceStepIdx, setInferenceStepIdx] = useState<number>(0);
   const [imageTitle, setImageTitle] = useState<string | null>(null);
   const [image, setImage] = useState<string | null>(null);
-  const [steps, setSteps] = useState<number>(10);
+  const [steps, setSteps] = useState<number>(40);
   const [showTextInput, setShowTextInput] = useState(false);
   const [keyboardVisible, setKeyboardVisible] = useState(false);
 
@@ -115,6 +115,8 @@ export default function TextToImageScreen() {
         <View style={styles.bottomContainer}>
           <BottomBarWithTextInput
             runModel={runForward}
+            numSteps={steps}
+            setSteps={setSteps}
             stopModel={model.interrupt}
             isGenerating={model.isGenerating}
             isReady={model.isReady}
diff --git a/apps/computer-vision/components/BottomBarWithTextInput.tsx b/apps/computer-vision/components/BottomBarWithTextInput.tsx
index 2ebf78a56..8cffec395 100644
--- a/apps/computer-vision/components/BottomBarWithTextInput.tsx
+++ b/apps/computer-vision/components/BottomBarWithTextInput.tsx
@@ -14,6 +14,8 @@ import ColorPalette from '../colors';
 interface BottomBarProps {
   runModel: (input: string, numSteps: number) => void;
   stopModel: () => void;
+  numSteps: number;
+  setSteps: React.Dispatch<React.SetStateAction<number>>;
   isGenerating?: boolean;
   isReady?: boolean;
   showTextInput: boolean;
@@ -24,6 +26,8 @@ interface BottomBarProps {
 export const BottomBarWithTextInput = ({
   runModel,
   stopModel,
+  numSteps,
+  setSteps,
   isGenerating,
   isReady,
   showTextInput,
@@ -31,10 +35,9 @@ export const BottomBarWithTextInput = ({
   keyboardVisible,
 }: BottomBarProps) => {
   const [input, setInput] = useState('');
-  const [numSteps, setNumSteps] = useState(10);
 
-  const decreaseSteps = () => setNumSteps((prev) => Math.max(5, prev - 5));
-  const increaseSteps = () => setNumSteps((prev) => Math.min(50, prev + 5));
+  const decreaseSteps = () => setSteps((prev) => Math.max(5, prev - 5));
+  const increaseSteps = () => setSteps((prev) => Math.min(50, prev + 5));
 
   if (!showTextInput) {
     if (isGenerating) {