Update app.py

app.py

@@ -247,7 +247,7 @@ def update_pricing(steps):
     cost = round(cost_per_second * total_seconds, 2)
     cost_preview = f'''To train this LoRA, a paid L4 GPU will be hooked under the hood during training and then removed once finished.
     ## Estimated to cost <b>< US$ {str(cost)}</b> for {round(int(total_seconds)/60, 2)} minutes with your current train settings <small>({int(steps)} iterations at {seconds_per_iteration}s/it)</small>'''
-    return gr.update(visible=True), gr.update(cost_preview)
+    return gr.update(visible=True), cost_preview
 
 with gr.Blocks(theme=theme, css=css) as demo:
     gr.Markdown(
@@ -388,6 +388,12 @@ with gr.Blocks(theme=theme, css=css) as demo:
         outputs=[cost_preview, cost_preview_info]
     )
 
+    steps.change(
+        update_pricing,
+        inputs=[steps],
+        outputs=[cost_preview, cost_preview_info]
+    )
+    
     start.click(fn=create_dataset, inputs=[images] + caption_list, outputs=dataset_folder).then(
         fn=start_training,
         inputs=[