Update app.py

app.py

@@ -55,14 +55,30 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer],  # Add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,
     planning_interval=None,
     name=None,
     description=None,
-    prompt_templates=prompt_templates
+    prompt_templates=prompt_templates,
+    additional_authorized_imports=[  
+        "unicodedata", "stat", "datetime", "random", "pandas", "numpy",
+        "itertools", "math", "statistics", "queue", "time", "collections",
+        "re", "os", "sys", "json", "joblib", "pickle", "gzip", "shutil", "pathlib",
+        
+        # Machine Learning & Deep Learning
+        "torch", "torchvision", "torchaudio", "torch.nn", "torch.optim",
+        "torch.utils.data", "tensorflow", "keras", "sklearn", "scipy", "cv2",
+        "transformers", "datasets", "tokenizers", "sentencepiece",
+
+        # Visualization
+        "matplotlib.pyplot", "seaborn", "plotly",
+
+        # Utility Libraries
+        "tqdm", "argparse", "logging"
+    ]
 )