Update app.py

app.py

@@ -4,19 +4,19 @@ from transformers import pipeline
 from langchain_community.llms import HuggingFacePipeline
 from langchain_experimental.agents import create_pandas_dataframe_agent
 
-# Load your data
+# Load data
 df = pd.read_csv("synthetic_profit.csv")
 
-# Setup lightweight Hugging Face pipeline with Flan-T5 base
+# Set up lightweight Hugging Face pipeline (Flan-T5 Base)
 hf_pipeline = pipeline(
     task="text2text-generation",
     model="google/flan-t5-base",
-    device=-1  # CPU only
+    device=-1  # CPU
 )
 
-# LangChain agent setup
+# LangChain agent setup (with explicit security opt-in)
 llm = HuggingFacePipeline(pipeline=hf_pipeline)
-agent = create_pandas_dataframe_agent(llm, df, verbose=True)
+agent = create_pandas_dataframe_agent(llm, df, verbose=True, allow_dangerous_code=True)
 
 def answer(query: str) -> str:
     try: