Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -2,6 +2,8 @@ from langchain_community.document_loaders import PyPDFLoader
|
|
| 2 |
from langchain_community.llms.OpenAI import OpenAI
|
| 3 |
import gradio as gr
|
| 4 |
from langchain.chains.summarize import load_summarize_chain
|
|
|
|
|
|
|
| 5 |
loader = PyPDFLoader("http://arxiv.org/pdf/2508.13246v1")
|
| 6 |
documents = loader.load()
|
| 7 |
llm = OpenAI(temperature=0)
|
|
@@ -9,7 +11,7 @@ llm = OpenAI(temperature=0)
|
|
| 9 |
def summarize_pdf (pdf_file_path, custom_prompt=""):
|
| 10 |
loader = PyPDFLoader(pdf_file_path)
|
| 11 |
docs = loader.load_and_split()
|
| 12 |
-
chain = load_summarize_chain(
|
| 13 |
summary = chain.run(docs)
|
| 14 |
|
| 15 |
return summary
|
|
|
|
| 2 |
from langchain_community.llms.OpenAI import OpenAI
|
| 3 |
import gradio as gr
|
| 4 |
from langchain.chains.summarize import load_summarize_chain
|
| 5 |
+
from huggingface_hub import InferenceClient
|
| 6 |
+
TEXT_MODEL_NAME = "google/gemma-3-270m"
|
| 7 |
loader = PyPDFLoader("http://arxiv.org/pdf/2508.13246v1")
|
| 8 |
documents = loader.load()
|
| 9 |
llm = OpenAI(temperature=0)
|
|
|
|
| 11 |
def summarize_pdf (pdf_file_path, custom_prompt=""):
|
| 12 |
loader = PyPDFLoader(pdf_file_path)
|
| 13 |
docs = loader.load_and_split()
|
| 14 |
+
chain = load_summarize_chain(TEXT_MODEL_NAME, chain_type="map_reduce")
|
| 15 |
summary = chain.run(docs)
|
| 16 |
|
| 17 |
return summary
|