Update app.py

app.py

@@ -184,15 +184,18 @@ def compare_models():
     return results_df
 
 def train_model(_):
-    wandb.login(key=os.environ.get("WANDB_API_KEY"))
-    run_counter = 1
-    # Then when initializing
-    wandb_run = wandb.init(project="huggingface-data-analysis", name=f"Optuna_Run_{run_counter}", reinit=True)
-    run_counter += 1
+    try:
+        wandb.login(key=os.environ.get("WANDB_API_KEY"))
+        run_counter = 1
+        wandb_run = wandb.init(
+            project="huggingface-data-analysis",
+            name=f"Optuna_Run_{run_counter}",
+            reinit=True
+        )
+        run_counter += 1
 
-      
         X_train, X_test, y_train, y_test = prepare_data()
-        
+
         def objective(trial):
             params = {
                 "n_estimators": trial.suggest_int("n_estimators", 50, 200),
@@ -236,7 +239,6 @@ def train_model(_):
         return {}, pd.DataFrame()
 
 
-
 def explainability(_):
     import warnings
     warnings.filterwarnings("ignore")