AIPretender commited on
Commit
9b96ec1
·
verified ·
1 Parent(s): 24eb5e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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(llm, chain_type="map_reduce")
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