Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,8 +27,9 @@ from langchain.llms import OpenAI
|
|
| 27 |
from langchain.document_loaders import PyPDFLoader
|
| 28 |
from langchain.indexes import VectorstoreIndexCreator
|
| 29 |
import tempfile
|
|
|
|
| 30 |
|
| 31 |
-
os.environ["OPENAI_API_KEY"] =
|
| 32 |
query1=" "
|
| 33 |
limit = 0
|
| 34 |
def loading_pdf():
|
|
@@ -36,11 +37,14 @@ def loading_pdf():
|
|
| 36 |
|
| 37 |
def pdf_changes(pdf_doc, prompt):
|
| 38 |
|
|
|
|
|
|
|
| 39 |
#loader = OnlinePDFLoader(pdf_doc.name)
|
| 40 |
-
loader = PyPDFLoader(pdf_doc.name)
|
| 41 |
-
documents =
|
|
|
|
| 42 |
|
| 43 |
-
name_filter = "**/*.md"
|
| 44 |
separator = "\n"
|
| 45 |
chunk_size_limit = 1000
|
| 46 |
max_chunk_overlap = 50
|
|
@@ -95,7 +99,7 @@ def bot(history):
|
|
| 95 |
def infer(question):
|
| 96 |
global query1
|
| 97 |
global limit
|
| 98 |
-
openai.api_key =
|
| 99 |
prompt_text = question
|
| 100 |
if prompt_text:
|
| 101 |
query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
|
|
@@ -128,7 +132,8 @@ with gr.Blocks(css=css,theme = gr.themes.Soft()) as demo:
|
|
| 128 |
#gr.HTML(title)
|
| 129 |
with gr.Row():
|
| 130 |
with gr.Column(scale=1):
|
| 131 |
-
|
|
|
|
| 132 |
prompt = gr.Textbox(label="Behaviour Prompt (optional)", placeholder="Reply to all questions as a rap / Reply to all questions in Hindi etc. ")
|
| 133 |
#repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz"], value="google/flan-ul2")
|
| 134 |
with gr.Row():
|
|
|
|
| 27 |
from langchain.document_loaders import PyPDFLoader
|
| 28 |
from langchain.indexes import VectorstoreIndexCreator
|
| 29 |
import tempfile
|
| 30 |
+
from langchain.document_loaders import DirectoryLoader
|
| 31 |
|
| 32 |
+
os.environ["OPENAI_API_KEY"] = "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"
|
| 33 |
query1=" "
|
| 34 |
limit = 0
|
| 35 |
def loading_pdf():
|
|
|
|
| 37 |
|
| 38 |
def pdf_changes(pdf_doc, prompt):
|
| 39 |
|
| 40 |
+
loader1 = DirectoryLoader(pdf_doc[0].name)
|
| 41 |
+
loader2 = DirectoryLoader(pdf_doc[1].name)
|
| 42 |
#loader = OnlinePDFLoader(pdf_doc.name)
|
| 43 |
+
#loader = PyPDFLoader(pdf_doc.name)
|
| 44 |
+
documents = loader1.load() + loader2.load()
|
| 45 |
+
print(len(documents))
|
| 46 |
|
| 47 |
+
#name_filter = "**/*.md"
|
| 48 |
separator = "\n"
|
| 49 |
chunk_size_limit = 1000
|
| 50 |
max_chunk_overlap = 50
|
|
|
|
| 99 |
def infer(question):
|
| 100 |
global query1
|
| 101 |
global limit
|
| 102 |
+
openai.api_key = "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"
|
| 103 |
prompt_text = question
|
| 104 |
if prompt_text:
|
| 105 |
query1 = query1 + "\nUser: " + prompt_text + "\nBot: "
|
|
|
|
| 132 |
#gr.HTML(title)
|
| 133 |
with gr.Row():
|
| 134 |
with gr.Column(scale=1):
|
| 135 |
+
#gr.File(file_count="multiple")
|
| 136 |
+
pdf_doc = gr.File(label="Load a pdf", file_count="multiple")
|
| 137 |
prompt = gr.Textbox(label="Behaviour Prompt (optional)", placeholder="Reply to all questions as a rap / Reply to all questions in Hindi etc. ")
|
| 138 |
#repo_id = gr.Dropdown(label="LLM", choices=["google/flan-ul2", "OpenAssistant/oasst-sft-1-pythia-12b", "bigscience/bloomz"], value="google/flan-ul2")
|
| 139 |
with gr.Row():
|