shroogawh2 commited on
Commit
e6a0569
·
verified ·
1 Parent(s): 658b0f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -9
app.py CHANGED
@@ -7,7 +7,6 @@ import openai
7
  import langchain
8
  import tiktoken
9
  from sentence_transformers import SentenceTransformer, util
10
- from langchain.embeddings import HuggingFaceBgeEmbeddings
11
  import gradio as gr
12
 
13
  openai.api_key ="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"
@@ -15,14 +14,7 @@ openai.api_key ="sk-proj-UPLtaXRZOgpqXhQC7aGBfQdah-xj4Wz0kmSpQ6r0r6CfdiTsL5FDiJU
15
  #documents=loader.load()
16
  #text_splitter = RecursiveCharacterTextSplitter(chunk_size = 400, chunk_overlap = 100)
17
  #docs = text_splitter.split_documents(documents=documents)
18
- model_name = "BAAI/bge-large-en-v1.5"
19
- model_kwargs = {'device':'cuda'}
20
- encode_kwargs = {'normalize_embeddings':True}
21
-
22
- embedding_function = HuggingFaceBgeEmbeddings(
23
- model_name = model_name,
24
- model_kwargs = model_kwargs,
25
- encode_kwargs = encode_kwargs)
26
 
27
  db = FAISS.load_local("Faiss",embedding_function, allow_dangerous_deserialization=True)
28
  parser = StrOutputParser()
 
7
  import langchain
8
  import tiktoken
9
  from sentence_transformers import SentenceTransformer, util
 
10
  import gradio as gr
11
 
12
  openai.api_key ="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"
 
14
  #documents=loader.load()
15
  #text_splitter = RecursiveCharacterTextSplitter(chunk_size = 400, chunk_overlap = 100)
16
  #docs = text_splitter.split_documents(documents=documents)
17
+ embedding_function = SentenceTransformer("BAAI/bge-large-en-v1.5")
 
 
 
 
 
 
 
18
 
19
  db = FAISS.load_local("Faiss",embedding_function, allow_dangerous_deserialization=True)
20
  parser = StrOutputParser()