Update app.py

app.py

@@ -76,7 +76,7 @@ class CustomTrainerCallback(TrainerCallback):
                 job.progress = 0.1 + (prog * 0.8)
                 if state.global_step % 1 == 0:
                     loss = state.log_history[-1].get('loss', 'N/A') if state.log_history else '...'
-                    job.add_log(f"Step {state.global_step}/{state.max_steps} | Loss: {loss}")
+                    job.add_log(f"Training Step {state.global_step}/{state.max_steps} | Loss: {loss}")
         return control
 
     def on_save(self, args, state, control, **kwargs):
@@ -236,7 +236,7 @@ def background_train_task(job_id, hf_token, model_name, new_repo_name, lora_r, l
             report_to="none",
             fp16=True if torch.cuda.is_available() else False,
             disable_tqdm=True,
-            dataloader_num_workers=cpu_count,
+            dataloader_num_workers=4,
             dataloader_pin_memory=True,
             torch_compile=True if os.name == 'posix' else False
         )
@@ -361,7 +361,7 @@ def load_from_url(request: gr.Request):
         pass
     return gr.update(selected="launch_tab"), ""
 
-with gr.Blocks(title="Nucleus Enterprise"):
+with gr.Blocks(title="Nucleus Enterprise") as demo:
     with gr.Column():
         gr.Markdown("# ⚛️ NUCLEUS ENTERPRISE")
         gr.Markdown("Autonomous LLM Foundry | V5.0 Stable")