Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,8 @@ from langchain.document_loaders import UnstructuredPowerPointLoader
|
|
19 |
# from langchain.document_loaders.image import UnstructuredImageLoader
|
20 |
|
21 |
|
22 |
-
|
|
|
23 |
|
24 |
from sentence_transformers import SentenceTransformer, models, util
|
25 |
word_embedding_model = models.Transformer('sentence-transformers/all-MiniLM-L6-v2', do_lower_case=True)
|
@@ -28,6 +29,7 @@ embedder = SentenceTransformer(modules=[word_embedding_model, pooling_model])
|
|
28 |
ocr = CnOcr()
|
29 |
# chat_url = 'https://Raghav001-API.hf.space/sale'
|
30 |
chat_url = 'https://Raghav001-API.hf.space/chatpdf'
|
|
|
31 |
headers = {
|
32 |
'Content-Type': 'application/json',
|
33 |
}
|
@@ -254,8 +256,13 @@ def pine(data):
|
|
254 |
c = c+1
|
255 |
|
256 |
|
257 |
-
Embedding_model = "text-embedding-ada-002"
|
258 |
-
embeddings = OpenAIEmbeddings(openai_api_key="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")
|
|
|
|
|
|
|
|
|
|
|
259 |
|
260 |
pinecone.init(api_key = "ffb1f594-0915-4ebf-835f-c1eaa62fdcdc",
|
261 |
environment = "us-west4-gcp-free"
|
@@ -266,7 +273,13 @@ def pine(data):
|
|
266 |
|
267 |
return ''
|
268 |
|
269 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
270 |
|
271 |
with gr.Blocks() as demo:
|
272 |
with gr.Row():
|
|
|
19 |
# from langchain.document_loaders.image import UnstructuredImageLoader
|
20 |
|
21 |
|
22 |
+
from langchain.chains.question_answering import load_qa_chain
|
23 |
+
from langchain import OpenAI
|
24 |
|
25 |
from sentence_transformers import SentenceTransformer, models, util
|
26 |
word_embedding_model = models.Transformer('sentence-transformers/all-MiniLM-L6-v2', do_lower_case=True)
|
|
|
29 |
ocr = CnOcr()
|
30 |
# chat_url = 'https://Raghav001-API.hf.space/sale'
|
31 |
chat_url = 'https://Raghav001-API.hf.space/chatpdf'
|
32 |
+
chat_emd = 'https://Raghav001-API.hf.space/embedd'
|
33 |
headers = {
|
34 |
'Content-Type': 'application/json',
|
35 |
}
|
|
|
256 |
c = c+1
|
257 |
|
258 |
|
259 |
+
# Embedding_model = "text-embedding-ada-002"
|
260 |
+
# embeddings = OpenAIEmbeddings(openai_api_key="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")
|
261 |
+
|
262 |
+
embeddings = requests.post(url=chat_emd,
|
263 |
+
data=json.dumps(data),
|
264 |
+
headers=headers
|
265 |
+
)
|
266 |
|
267 |
pinecone.init(api_key = "ffb1f594-0915-4ebf-835f-c1eaa62fdcdc",
|
268 |
environment = "us-west4-gcp-free"
|
|
|
273 |
|
274 |
return ''
|
275 |
|
276 |
+
def get_answer(query_live):
|
277 |
+
|
278 |
+
llm = OpenAI(temperature=0, openai)
|
279 |
+
qa_chain = load_qa_chain(llm,chain_type='stuff')
|
280 |
+
query = query_live
|
281 |
+
docs = docstore.similarity_search(query)
|
282 |
+
qa_chain.run(input_documents = docs, question = query)
|
283 |
|
284 |
with gr.Blocks() as demo:
|
285 |
with gr.Row():
|