Update app.py

app.py

@@ -533,7 +533,7 @@ def build_ui():
     if not models:
         models = list(MODEL_CATALOG.keys())[:2]  # fallback
 
-    with gr.Blocks(css=CSS, title="Fine-Tuning Studio") as demo:
+    with gr.Blocks(title="Fine-Tuning Studio") as demo:
 
         # ── HEADER ────────────────────────────────────────────
         gr.HTML(f"""
@@ -650,13 +650,11 @@ def build_ui():
                     return "⏹ Cancelamento solicitado."
 
                 def poll_logs():
-                    """Polling a cada 2s para atualizar log e progresso."""
-                    while True:
-                        pct = 0
-                        if STATE.total_steps > 0:
-                            pct = min(100, int(STATE.progress / STATE.total_steps * 100))
-                        yield STATE.log_box(), pct
-                        time.sleep(2)
+                    """Retorna snapshot atual do log e progresso (chamado pelo Timer)."""
+                    pct = 0
+                    if STATE.total_steps > 0:
+                        pct = min(100, int(STATE.progress / STATE.total_steps * 100))
+                    return STATE.log_box(), pct
 
                 train_btn.click(
                     start_training,
@@ -670,7 +668,8 @@ def build_ui():
                 )
                 cancel_btn.click(cancel_training, outputs=status_md)
 
-                demo.load(poll_logs, outputs=[log_box, progress], every=2)
+                # Gradio 6.0: gr.Timer substitui every= no demo.load
+                gr.Timer(value=2).tick(poll_logs, outputs=[log_box, progress])
 
             # ════════════════════════════════
             #  TAB 2 — CHAT
@@ -773,4 +772,5 @@ if __name__ == "__main__":
         server_port=7860,
         share=False,
         show_error=True,
+        css=CSS,
     )