varl42 commited on
Commit
f44133f
1 Parent(s): 9e3d236

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -17,7 +17,7 @@ This application summarizes PDF documents and converts the summary to audio.
17
  The core logic is in the `audio_pdf` function. It:
18
 
19
  1. Extracts raw text from the uploaded PDF using `PyPDF2`
20
- 2. Summarizes the text using a BART model from HuggingFace Transformers. This uses `AutoTokenizer` and `AutoModelForSeq2SeqLM` to load the model and generate a summary
21
  3. Converts the text summary to an audio file using `gTTS` (Google Text-to-Speech)
22
 
23
  The summary and audio file are returned and displayed in the Gradio web interface.
 
17
  The core logic is in the `audio_pdf` function. It:
18
 
19
  1. Extracts raw text from the uploaded PDF using `PyPDF2`
20
+ 2. Summarizes the text using [LED-Based Summarization](https://huggingface.co/pszemraj/led-base-book-summary) Model from HuggingFace Transformers. This uses `AutoTokenizer` and `AutoModelForSeq2SeqLM` to load the model and generate a summary
21
  3. Converts the text summary to an audio file using `gTTS` (Google Text-to-Speech)
22
 
23
  The summary and audio file are returned and displayed in the Gradio web interface.