Update app.py

app.py

@@ -81,11 +81,6 @@ def format_insights(insights, visuals):
 
 def analyze_data(csv_file, additional_notes=""):
 
-def objective(trial):
-    learning_rate = trial.suggest_loguniform("learning_rate", 1e-5, 5e-3)
-    batch_size = trial.suggest_categorical("batch_size", [8, 16, 32])
-    num_epochs = trial.suggest_int("num_epochs", 1, 5)
-
 
        # Start timing
     start_time = time.time()
@@ -146,7 +141,11 @@ def objective(trial):
         """,
         additional_args={}
     )
-
+def objective(trial):
+    learning_rate = trial.suggest_loguniform("learning_rate", 1e-5, 5e-3)
+    batch_size = trial.suggest_categorical("batch_size", [8, 16, 32])
+    num_epochs = trial.suggest_int("num_epochs", 1, 5)
+    
 def tune_hyperparameters(n_trials: int):
     study = optuna.create_study(direction="minimize")
     study.optimize(objective, n_trials=n_trials)