Update app.py

app.py

@@ -85,7 +85,7 @@ class CustomTrainerCallback(TrainerCallback):
             ckpt_name = f"checkpoint-{step}"
             ckpt_path = os.path.join(args.output_dir, ckpt_name)
             
-            job.add_log(f"System: Full State Snapshot saved ({ckpt_name})")
+            job.add_log(f"System: 100-Step Snapshot saved ({ckpt_name})")
             
             def _upload_bg():
                 try:
@@ -94,9 +94,9 @@ class CustomTrainerCallback(TrainerCallback):
                         path_in_repo=".",
                         repo_id=self.repo_id,
                         token=self.hf_token,
-                        commit_message=f"Live Full Checkpoint {step}"
+                        commit_message=f"Live Checkpoint Step {step}"
                     )
-                    job.add_log(f"Cloud: Synced Full Checkpoint {step} to Root")
+                    job.add_log(f"Cloud: Synced Checkpoint {step} to Root")
                 except:
                     pass
 
@@ -221,7 +221,7 @@ def background_train_task(job_id, hf_token, model_name, new_repo_name,
             optim="adamw_torch_fused" if torch.cuda.is_available() else "adamw_torch",
             logging_steps=1,
             save_strategy="steps",
-            save_steps=max(5, int(int(train_steps)/5)),
+            save_steps=100,
             save_total_limit=1,
             report_to="none",
             fp16=True if torch.cuda.is_available() else False,
@@ -343,7 +343,7 @@ def load_from_url(request: gr.Request):
         pass
     return gr.update(selected="launch_tab"), ""
 
-with gr.Blocks(title="Nucleus Enterprise") as demo:
+with gr.Blocks(title="Nucleus Enterprise", theme=gr.themes.Base()) as demo:
     with gr.Column():
         gr.Markdown("# ⚛️ NUCLEUS ENTERPRISE")
         gr.Markdown("Autonomous LLM Foundry | V7.0 Scratch Edition")