Dua Rajper commited on
Commit
d348825
·
verified ·
1 Parent(s): 2ae5ce8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ from langchain.embeddings import HuggingFaceEmbeddings
8
  from langchain.vectorstores import FAISS
9
  from langchain.memory import ConversationBufferMemory
10
  from langchain.chains import ConversationalRetrievalChain
11
- from langchain.llms import HuggingFaceHub
12
 
13
  # Load environment variables
14
  load_dotenv()
@@ -42,7 +42,7 @@ def get_vectorstore(text_chunks):
42
 
43
  def get_conversation_chain(vectorstore):
44
  try:
45
- llm = HuggingFaceHub(repo_id="mistralai/Mistral-7B-Instruct-v0.1", model_kwargs={"temperature": 0.5})
46
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
47
 
48
  return ConversationalRetrievalChain.from_llm(
 
8
  from langchain.vectorstores import FAISS
9
  from langchain.memory import ConversationBufferMemory
10
  from langchain.chains import ConversationalRetrievalChain
11
+ from langchain.llms import OpenAI
12
 
13
  # Load environment variables
14
  load_dotenv()
 
42
 
43
  def get_conversation_chain(vectorstore):
44
  try:
45
+ llm = OpenAI(model_name="gpt-4", api_key=os.getenv("OPENAI_API_KEY"))
46
  memory = ConversationBufferMemory(memory_key='chat_history', return_messages=True)
47
 
48
  return ConversationalRetrievalChain.from_llm(