CosmoAI commited on
Commit
e648470
·
1 Parent(s): 0182110

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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 OpenAIEmbeddings, HuggingFaceInstructEmbeddings
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