Update agent.py

agent.py

@@ -41,7 +41,7 @@ from io import StringIO
 
 from transformers import BertTokenizer, BertModel
 import torch
-#from langchain.embeddings import Embedding
+from langchain.embeddings import Embedding
 
 
 load_dotenv()
@@ -320,10 +320,7 @@ for task in tasks:
 #embedding_model = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-base-v2")
 #embedding_model = HuggingFaceEmbeddings(model_name="BAAI/bge-base-en-v1.5")
 
-from transformers import BertTokenizer, BertModel
-import torch
-from langchain.embeddings import Embedding
-from langchain.schema import Document
+
 
 class BERTEmbedding(Embedding):
     def __init__(self, model_name='bert-base-uncased'):