Raghav001 commited on
Commit
20b8359
·
1 Parent(s): c5572b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -4
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():