Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,14 +2,14 @@ import streamlit as st
|
|
2 |
from dotenv import load_dotenv
|
3 |
from PyPDF2 import PdfReader
|
4 |
from langchain.text_splitter import CharacterTextSplitter
|
5 |
-
from langchain.embeddings import
|
6 |
from langchain.vectorstores import FAISS
|
7 |
-
from langchain.chat_models import ChatOpenAI
|
8 |
from langchain.memory import ConversationBufferMemory
|
9 |
from langchain.chains import ConversationalRetrievalChain
|
10 |
from htmlTemplates import css, bot_template, user_template
|
11 |
from langchain.llms import HuggingFaceHub
|
12 |
-
from langchain.callbacks import get_openai_callback
|
13 |
|
14 |
def get_pdf_text(pdf_docs):
|
15 |
text = ""
|
@@ -33,7 +33,8 @@ def get_text_chunks(text):
|
|
33 |
|
34 |
def get_vectorstore(text_chunks):
|
35 |
# embeddings = OpenAIEmbeddings()
|
36 |
-
embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-xl")
|
|
|
37 |
vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
|
38 |
return vectorstore
|
39 |
|
|
|
2 |
from dotenv import load_dotenv
|
3 |
from PyPDF2 import PdfReader
|
4 |
from langchain.text_splitter import CharacterTextSplitter
|
5 |
+
from langchain.embeddings import HuggingFaceEmbeddings
|
6 |
from langchain.vectorstores import FAISS
|
7 |
+
# from langchain.chat_models import ChatOpenAI
|
8 |
from langchain.memory import ConversationBufferMemory
|
9 |
from langchain.chains import ConversationalRetrievalChain
|
10 |
from htmlTemplates import css, bot_template, user_template
|
11 |
from langchain.llms import HuggingFaceHub
|
12 |
+
# from langchain.callbacks import get_openai_callback
|
13 |
|
14 |
def get_pdf_text(pdf_docs):
|
15 |
text = ""
|
|
|
33 |
|
34 |
def get_vectorstore(text_chunks):
|
35 |
# embeddings = OpenAIEmbeddings()
|
36 |
+
# embeddings = HuggingFaceInstructEmbeddings(model_name="hkunlp/instructor-xl")
|
37 |
+
embeddings = HuggingFaceEmbeddings()
|
38 |
vectorstore = FAISS.from_texts(texts=text_chunks, embedding=embeddings)
|
39 |
return vectorstore
|
40 |
|