Update app.py

app.py

@@ -1,4 +1,5 @@
 import os, re, requests, pandas as pd, gradio as gr
+from transformers import pipeline
 from langchain_huggingface.llms import HuggingFacePipeline
 from langchain.tools import tool
 from langchain_core.output_parsers import JsonOutputParser
@@ -132,13 +133,8 @@ class BasicAgent:
         # initialize HF inference pipeline once
         if HF_TOKEN is None:
             raise ValueError("HF_TOKEN not set in environment")
-        self.llm = HuggingFacePipeline(
-            model="distilbert/distilgpt2",
-            tokenizer="distilbert/distilgpt2",
-            task="text-generation",
-            max_new_tokens=16,
-            token=HF_TOKEN
-        ).bind_tools(TOOLS)
+        pipe = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", max_new_tokens=16)
+        self.llm = HuggingFacePipeline(pipeline=pipe).bind_tools(TOOLS)
         # The GAIA system prompt (no "FINAL ANSWER:" at the end)
         self.system_prompt = SYSTEM_MESSAGE
         print("BasicAgent initialized with LLM.")