Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -54,6 +54,8 @@ from huggingface_hub import login
|
|
| 54 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
|
| 55 |
from langchain_huggingface import HuggingFaceEndpoint
|
| 56 |
from langchain.agents import initialize_agent
|
|
|
|
|
|
|
| 57 |
|
| 58 |
load_dotenv()
|
| 59 |
|
|
|
|
| 54 |
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
|
| 55 |
from langchain_huggingface import HuggingFaceEndpoint
|
| 56 |
from langchain.agents import initialize_agent
|
| 57 |
+
from langchain.agents import AgentType
|
| 58 |
+
|
| 59 |
|
| 60 |
load_dotenv()
|
| 61 |
|