Update app.py

app.py

@@ -5,7 +5,7 @@ import json
 from langchain.document_loaders import DataFrameLoader
 from langchain.text_splitter import CharacterTextSplitter
 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
 
@@ -39,11 +39,7 @@ def url_changes(url, pages_to_visit, urls_to_scrape, repo_id):
     documents = loader.load()
     text_splitter = CharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
     texts = text_splitter.split_documents(documents)
-    #"snunlp/KR-SBERT-V40K-klueNLI-augSTS"
-    emb_model = "all-MiniLM-L6-v2"
-    embeddings = HuggingFaceEmbeddings(
-        model_name=emb_model
-    )
+    embeddings = SentenceTransformerEmbeddings(model_name="jhgan/ko-sroberta-multitask")
     db = Chroma.from_documents(texts, embeddings)
     retriever = db.as_retriever()
     llm = HuggingFaceHub(repo_id=repo_id, model_kwargs={"temperature":0.1, "max_new_tokens":250})