Update agent.py

agent.py

@@ -65,8 +65,7 @@ from functools import reduce
 from youtube_transcript_api import YouTubeTranscriptApi
 from youtube_transcript_api._errors import TranscriptsDisabled, VideoUnavailable
 import faiss
-from langchain.vectorstores import FAISS
-from langchain.embeddings import BERTEmbeddings
+
 from langchain.schema import Document
 
 
@@ -420,7 +419,7 @@ docs = [
 texts = [doc.page_content for doc in docs]
 
 # Initialize the embedding model
-embedding_model = BERTEmbeddings()
+embedding_model = HuggingFaceEmbeddings(model_name="bert-base-uncased")
 
 embeddings = [embedding_model.embed_query(text) for text in texts]