ajibs75 commited on
Commit
24ccd6f
·
verified ·
1 Parent(s): a5d7ba2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ app.add_middleware(
22
 
23
  # Initialize the summarization pipeline
24
  device = "cuda" if torch.cuda.is_available() else "cpu"
25
- summarization_pipe = pipeline("summarization", model="Falconsai/text_summarization", device=device,cache_dir="cache" )
26
 
27
  def extract_text_from_pdf(file_bytes):
28
  pdf_reader = PyPDF2.PdfReader(io.BytesIO(file_bytes))
 
22
 
23
  # Initialize the summarization pipeline
24
  device = "cuda" if torch.cuda.is_available() else "cpu"
25
+ summarization_pipe = pipeline("summarization", model="Falconsai/text_summarization", device=device )
26
 
27
  def extract_text_from_pdf(file_bytes):
28
  pdf_reader = PyPDF2.PdfReader(io.BytesIO(file_bytes))