Update app.py

app.py

@@ -10,6 +10,8 @@ from langchain.tools import DuckDuckGoSearchRun
 from langchain_community.llms import HuggingFaceHub
 from langchain_huggingface import HuggingFaceEndpoint
 from langchain_community.chat_models import ChatHuggingFace
+from langchain.llms.base import LLM
+from huggingface_hub import InferenceClient
 
 # (Keep Constants as is)
 # --- Constants ---
@@ -44,7 +46,7 @@ class BasicAgent: # Some times Inheritance is needed
         llm = ChatHuggingFace(
             repo_id="tiiuae/falcon-7b-instruct",
             huggingfacehub_api_token=os.environ["HUGGINGFACEHUB_API_TOKEN"],
-            model_kwargs={"temperature": 0.5, "max_new_tokens": 512}
+            llm={"temperature": 0.5, "max_new_tokens": 512}
         )
         
         # # Define the LLM