Update app.py

app.py

@@ -5,7 +5,7 @@ import json
 from langchain.document_loaders import DataFrameLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain.llms import HuggingFaceHub
-from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.embeddings.sentence_transformer import SentenceTransformerEmbeddings
 from langchain.vectorstores import Chroma
 from langchain.chains import RetrievalQA
 
@@ -45,11 +45,12 @@ def url_changes(url, pages_to_visit, urls_to_scrape, repo_id):
     texts = text_splitter.split_documents(documents)
     print(f"documents splitted into {len(texts)} chunks") 
     
-    embeddings = HuggingFaceEmbeddings(model_name="jhgan/ko-sroberta-multitask")
+    embeddings = SentenceTransformerEmbeddings(model_name="jhgan/ko-sroberta-multitask")
 
     persist_directory = './vector_db'
     db = Chroma.from_documents(texts, embeddings, persist_directory=persist_directory)
     retriever = db.as_retriever()
+    
     llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature":0.1, "max_new_tokens":250})
     global qa
     qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff", retriever=retriever, return_source_documents=True)