Update app.py

app.py

@@ -9,6 +9,8 @@ from langchain.agents import Tool, AgentExecutor, initialize_agent
 from langchain.agents.agent_types import AgentType
 from langchain.prompts.chat import SystemMessagePromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate
 from langchain_huggingface import HuggingFaceEndpoint
+from huggingface_hub import login
+from langchain.llms import HuggingFaceHub
 
 
 
@@ -52,7 +54,7 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
         login(token=os.getenv('HF_TOKEN'))
 
 # Use a Hugging Face cloud model
-        llm = HuggingFaceEndpoint(
+        llm = HuggingFaceHub(
             repo_id="google/flan-t5-base",  # Example cloud model; change as needed, 
             task="text-generation"
         )