Update app.py

app.py

@@ -1,17 +1,11 @@
-from transformers import pipeline
-import os
-import re
-import torch
-import gradio as gr
-import requests
-import inspect
-import pandas as pd
+import os, re, requests, pandas as pd, gradio as gr
 from langchain_huggingface.llms import HuggingFacePipeline
 from langchain_core.tools import tool
 from langchain_core.agents import AgentExecutor, JsonOutputParser
 from youtube_transcript_api import YouTubeTranscriptApi
 import chess, chess.engine
 from bs4 import BeautifulSoup
+import statsapi
 from SPARQLWrapper import SPARQLWrapper, JSON
 
 # (Keep Constants as is)
@@ -192,9 +186,13 @@ class BasicAgent:
         # initialize HF inference pipeline once
         if HF_TOKEN is None:
             raise ValueError("HF_TOKEN not set in environment")
-        self.generator = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", max_new_tokens=16)
-        self.llm = HuggingFacePipeline.from_pipeline(self.generator)
-        self.llm = self.llm.bind_tools(TOOLS)
+        self.llm = HuggingFacePipeline(
+            model="distilbert/distilgpt2",
+            tokenizer="distilbert/distilgpt2",
+            task="text-generation",
+            max_new_tokens=16,
+            token=HF_TOKEN
+        ).bind_tools(TOOLS)
         # The GAIA system prompt (no "FINAL ANSWER:" at the end)
         self.system_prompt = SYSTEM_MESSAGE
         print("BasicAgent initialized with LLM.")