Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -452,7 +452,11 @@ tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, vi
|
|
452 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
|
453 |
#llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
|
454 |
|
|
|
|
|
|
|
455 |
from transformers import pipeline
|
|
|
456 |
|
457 |
# Get the Hugging Face API token from the environment variable
|
458 |
# Get the Hugging Face API token from the environment variable
|
|
|
452 |
#llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo") # or "gpt-3.5-turbo" "gpt-4"
|
453 |
#llm = ChatMistralAI(model="mistral-7b-instruct-v0.1")
|
454 |
|
455 |
+
from langchain.agents import initialize_agent, Tool, AgentType
|
456 |
+
from langchain_huggingface import HuggingFaceHub
|
457 |
+
from langchain.prompts import PromptTemplate
|
458 |
from transformers import pipeline
|
459 |
+
from langchain_community.llms import HuggingFaceHub
|
460 |
|
461 |
# Get the Hugging Face API token from the environment variable
|
462 |
# Get the Hugging Face API token from the environment variable
|