fix

app.py

@@ -238,15 +238,19 @@ class OptimizedGAIAAgent:
             get_detailed_wikipedia
         ]
         try:
-            model_name = "gpt2"  # Replace with larger model if needed
+            
+            model_name = "gpt2"  # Or any other model you want
+
             tokenizer = AutoTokenizer.from_pretrained(model_name)
             model = AutoModelForCausalLM.from_pretrained(model_name)
-            generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
+            # Pass the raw model and tokenizer (or just model) to CodeAgent
             self.agent = CodeAgent(
                 tools=self.tools,
-                model=generator  # ✅ Model object passed here
+                model=model,           # <-- raw model object, not pipeline
+                tokenizer=tokenizer    # (if CodeAgent accepts tokenizer separately)
             )
+
             print("✅ CodeAgent initialized with model object")
         except Exception as e:
             print(f"⚠️ CodeAgent failed: {e}")