isayahc commited on
Commit
449fa63
·
verified ·
1 Parent(s): 65d65e8

refactored code for readability

Browse files
Files changed (1) hide show
  1. app.py +13 -7
app.py CHANGED
@@ -4,32 +4,38 @@ from langchain.document_loaders import OnlinePDFLoader
4
 
5
  from langchain.text_splitter import CharacterTextSplitter
6
  from langchain.prompts import PromptTemplate
 
7
 
8
  # from langhchain.llms import openai
9
  from langchain.llms import OpenAI
10
 
 
 
 
 
 
 
 
 
 
 
11
  text_splitter = CharacterTextSplitter(chunk_size=350, chunk_overlap=0)
12
 
13
- from langchain.llms import HuggingFaceHub
14
  # flan_ul2 = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
15
  flan_ul2 = OpenAI()
16
 
17
  global qa
18
 
19
- from langchain.embeddings import HuggingFaceHubEmbeddings, OpenAIEmbeddings
20
  # embeddings = HuggingFaceHubEmbeddings()
21
- embeddings = OpenAIEmbeddings()
22
 
23
- from langchain.vectorstores import Chroma
24
 
25
- from langchain.chains import RetrievalQA
26
 
27
- from langchain.document_loaders import PyPDFLoader
28
 
29
  def loading_pdf():
30
  return "Loading..."
31
  def pdf_changes(pdf_doc):
32
  # loader = OnlinePDFLoader(pdf_doc.name)
 
33
  loader = PyPDFLoader(pdf_doc.name)
34
  documents = loader.load()
35
  texts = text_splitter.split_documents(documents)
@@ -42,7 +48,7 @@ def pdf_changes(pdf_doc):
42
 
43
  {context}
44
 
45
- Question: {sample.question}
46
  Answer:"""
47
  PROMPT = PromptTemplate(
48
  template=prompt_template, input_variables=["context", "question"]
 
4
 
5
  from langchain.text_splitter import CharacterTextSplitter
6
  from langchain.prompts import PromptTemplate
7
+ from langchain.llms import HuggingFaceHub
8
 
9
  # from langhchain.llms import openai
10
  from langchain.llms import OpenAI
11
 
12
+ from langchain.vectorstores import Chroma
13
+
14
+ from langchain.chains import RetrievalQA
15
+
16
+ from langchain.document_loaders import PyPDFLoader
17
+
18
+
19
+ from langchain.embeddings import HuggingFaceHubEmbeddings, OpenAIEmbeddings
20
+
21
+
22
  text_splitter = CharacterTextSplitter(chunk_size=350, chunk_overlap=0)
23
 
 
24
  # flan_ul2 = HuggingFaceHub(repo_id="HuggingFaceH4/zephyr-7b-beta", model_kwargs={"temperature":0.1, "max_new_tokens":300})
25
  flan_ul2 = OpenAI()
26
 
27
  global qa
28
 
 
29
  # embeddings = HuggingFaceHubEmbeddings()
 
30
 
 
31
 
 
32
 
 
33
 
34
  def loading_pdf():
35
  return "Loading..."
36
  def pdf_changes(pdf_doc):
37
  # loader = OnlinePDFLoader(pdf_doc.name)
38
+ embeddings = OpenAIEmbeddings()
39
  loader = PyPDFLoader(pdf_doc.name)
40
  documents = loader.load()
41
  texts = text_splitter.split_documents(documents)
 
48
 
49
  {context}
50
 
51
+ Question: {query}
52
  Answer:"""
53
  PROMPT = PromptTemplate(
54
  template=prompt_template, input_variables=["context", "question"]