Spaces:
Sleeping
Sleeping
Dua Rajper commited on
Update app.py
Browse files
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
|
| 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 =
|
| 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(
|