Spaces:
Sleeping
Sleeping
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() |