Update app.py

app.py

@@ -56,7 +56,7 @@ agent = CodeAgent(
 
 
 # Gradio Gallery and visualization output
-visual_output = gr.Gallery(label="Generated Visualizations", columns=3, height=400)
+visual_output = gr.Gallery(label="Generated Visualizations", columns=3, height=600, object_fit="contain")
 
 # Fix in the `run_agent` function to handle agent results correctly
 def run_agent(_):
@@ -100,17 +100,9 @@ def run_agent(_):
     return insights, image_paths
 
 
-
-
-
-
-
 def train_model(_):
     wandb.login(key=os.environ.get("WANDB_API_KEY"))
-    #wandb_run = wandb.init(project="huggingface-data-analysis", name="Optuna_Run", reinit=True)
-    # At the start of your script
     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