Update app.py

app.py

@@ -81,10 +81,13 @@ torch.cuda.empty_cache()
 #pip install wandb
 import os
 import wandb
+api_key = os.getenv("API_KEY")
+
+# Authenticate with WandB
+wandb.login(key=api_key)
 #print(os.getenv('API_KEY'))
-os.environ["API_KEY"]
+#os.environ["API_KEY"]
 
-wandb.login(key="API_KEY")
 from transformers import MT5ForConditionalGeneration, Seq2SeqTrainingArguments, Seq2SeqTrainer
 import torch