Final_Assignment_Project

Running

wt002 commited on 4 days ago

Commit

4d61251

verified ·

1 Parent(s): f1d6b50

Update agent.py

Files changed (1) hide show

agent.py CHANGED Viewed

@@ -53,7 +53,7 @@ from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
 from langchain.agents import initialize_agent, Tool, AgentType
 from transformers import pipeline
-from smolagents import LiteLLMModel
 load_dotenv()

 from langchain.chains import LLMChain
 from langchain.agents import initialize_agent, Tool, AgentType
 from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 load_dotenv()