sahanes commited on
Commit
fb04de4
·
verified ·
1 Parent(s): ce43eb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -19
app.py CHANGED
@@ -2,35 +2,26 @@
2
  import os
3
  import gradio as gr
4
  import pandas as pd
5
- #import pickle
6
- # from langchain.embeddings.openai import OpenAIEmbeddings
7
- #import openai
8
  from langchain_openai import OpenAIEmbeddings
9
  from langchain.text_splitter import CharacterTextSplitter
10
  from langchain.text_splitter import RecursiveCharacterTextSplitter
11
- from langchain_community.vectorstores import Chroma
12
  #from langchain.vectorstores import Chroma
 
13
  from langchain_community.document_loaders import TextLoader
14
  from langchain_community.chat_models import ChatOpenAI
15
  # from langchain import PromptTemplate
16
  from langchain_core.prompts import PromptTemplate
17
  from langchain.chains import LLMChain
18
  from langchain_community.llms import OpenAI
19
- from langchain_community.vectorstores.faiss import FAISS
20
  #from langchain.vectorstores import FAISS
21
-
22
 
23
  #export OPENAI_API_KEY='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'
24
 
25
  # Set the OPENAI_API_KEY environment variable directly
26
  os.environ["OPENAI_API_KEY"] = "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"
27
 
28
- # db = faiss.read_index('index.pkl')
29
- #db = pickle.load('index.pkl')
30
- # with open('index.pickle', 'rb') as pkl:
31
- # doc_embedding = pickle.load(pkl)
32
- # db.save_local("faiss_index")
33
-
34
  # embeddings = OpenAIEmbeddings()
35
  # OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
36
 
@@ -43,14 +34,9 @@ try:
43
  except ValidationError as e:
44
  print(f"Error: {e}")
45
 
46
- # Load the FAISS index from the .pkl file
47
- #with open("index.pkl", "rb") as f:
48
- # db = faiss.read_index(f.read())
49
- #with open("index.pkl", "rb") as f:
50
- #db = faiss.deserialize_index(f.read())
51
 
52
  #------------------------------------------------------------------------------
53
- db=FAISS.load_local("faiss_index/faiss_index", embeddings,allow_dangerous_deserialization=True)
54
  #-----------------------------------------------------------------------------
55
  def get_response_from_query(db, query, k=3):
56
 
@@ -58,7 +44,6 @@ def get_response_from_query(db, query, k=3):
58
 
59
  docs_page_content = " ".join([d.page_content for d in docs])
60
 
61
- # llm = BardLLM()
62
  llm = ChatOpenAI(model_name="gpt-3.5-turbo-16k",temperature=0)
63
 
64
  prompt = PromptTemplate(
 
2
  import os
3
  import gradio as gr
4
  import pandas as pd
5
+
 
 
6
  from langchain_openai import OpenAIEmbeddings
7
  from langchain.text_splitter import CharacterTextSplitter
8
  from langchain.text_splitter import RecursiveCharacterTextSplitter
 
9
  #from langchain.vectorstores import Chroma
10
+ from langchain_community.vectorstores import Chroma
11
  from langchain_community.document_loaders import TextLoader
12
  from langchain_community.chat_models import ChatOpenAI
13
  # from langchain import PromptTemplate
14
  from langchain_core.prompts import PromptTemplate
15
  from langchain.chains import LLMChain
16
  from langchain_community.llms import OpenAI
 
17
  #from langchain.vectorstores import FAISS
18
+ from langchain_community.vectorstores.faiss import FAISS
19
 
20
  #export OPENAI_API_KEY='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'
21
 
22
  # Set the OPENAI_API_KEY environment variable directly
23
  os.environ["OPENAI_API_KEY"] = "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"
24
 
 
 
 
 
 
 
25
  # embeddings = OpenAIEmbeddings()
26
  # OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
27
 
 
34
  except ValidationError as e:
35
  print(f"Error: {e}")
36
 
 
 
 
 
 
37
 
38
  #------------------------------------------------------------------------------
39
+ db=FAISS.load_local("faiss_index", embeddings,allow_dangerous_deserialization=True)
40
  #-----------------------------------------------------------------------------
41
  def get_response_from_query(db, query, k=3):
42
 
 
44
 
45
  docs_page_content = " ".join([d.page_content for d in docs])
46
 
 
47
  llm = ChatOpenAI(model_name="gpt-3.5-turbo-16k",temperature=0)
48
 
49
  prompt = PromptTemplate(