Jimmy0866 commited on
Commit
6bd0a38
·
verified ·
1 Parent(s): 65a88c6

Upload 2 files

Browse files
Files changed (2) hide show
  1. app.py +2 -1
  2. requirements.txt +9 -7
app.py CHANGED
@@ -36,6 +36,7 @@ logging.basicConfig(
36
 
37
  import gradio as gr
38
  import nltk
 
39
  import torch
40
  from cleantext import clean
41
  from doctr.models import ocr_predictor
@@ -663,4 +664,4 @@ if __name__ == "__main__":
663
  inputs=[summary_text, text_file],
664
  outputs=[aggregated_summary],
665
  )
666
- demo.launch(share=args.share, debug=True)
 
36
 
37
  import gradio as gr
38
  import nltk
39
+ nltk.download('punkt') # 自動下載 tokenizer 資源
40
  import torch
41
  from cleantext import clean
42
  from doctr.models import ocr_predictor
 
664
  inputs=[summary_text, text_file],
665
  outputs=[aggregated_summary],
666
  )
667
+ demo.launch(share=args.share, debug=True)
requirements.txt CHANGED
@@ -1,7 +1,9 @@
1
- transformers>=4.41.0
2
- sentencepiece
3
- torch
4
- pdf2image
5
- poppler-utils
6
- pytesseract
7
- Pillow
 
 
 
1
+ torch>=2.0.0
2
+ torchvision>=0.15.0
3
+ transformers>=4.46.0
4
+ gradio>=4.24.0
5
+ pdf2image
6
+ pytesseract
7
+ nltk
8
+ Pillow
9
+ numpy