Disable the use_wandb button when WANDB_API_KEY is not set

app_training.py

@@ -2,6 +2,8 @@
 
 from __future__ import annotations
 
+import os
+
 import gradio as gr
 
 from constants import UploadTarget
@@ -73,7 +75,10 @@ def create_training_demo(trainer: Trainer,
                 checkpointing_steps = gr.Number(label='Checkpointing Steps',
                                                 value=100,
                                                 precision=0)
-                use_wandb = gr.Checkbox(label='Use W&B', value=False)
+                use_wandb = gr.Checkbox(label='Use W&B',
+                                        value=False,
+                                        interactive=bool(
+                                            os.getenv('WANDB_API_KEY')))
                 validation_epochs = gr.Number(label='Validation Epochs',
                                               value=100,
                                               precision=0)