Spaces:
Build error
Build error
| import gradio as gr | |
| import os | |
| import openai | |
| from dotenv import load_dotenv, find_dotenv | |
| from langchain import OpenAI, PromptTemplate | |
| from langchain.chains.summarize import load_summarize_chain | |
| from langchain.document_loaders import PyPDFLoader | |
| _ = load_dotenv(find_dotenv()) | |
| openai.api_key = os.environ['OPENAI_API_KEY'] | |
| llm = OpenAI(temperature=0) | |
| def pdf_summarizer(pdf): | |
| loader = PyPDFLoader(pdf.name) | |
| docs = loader.load_and_split() | |
| chain = load_summarize_chain(llm, chain_type="map_reduce") | |
| summary = chain.run(docs) | |
| return summary | |
| inputs = gr.File(label="Upload only a PDF file") | |
| outputs = gr.Textbox(label="PDF Summary") | |
| iface = gr.Interface( | |
| fn=pdf_summarizer, | |
| inputs=inputs, | |
| outputs=outputs, | |
| title="PDF Summarizer", | |
| ) | |
| iface.launch() |