Kaiyeee commited on
Commit
0c7e107
·
verified ·
1 Parent(s): 2e0a452

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -4
app.py CHANGED
@@ -1,5 +1,3 @@
1
- # app.py
2
-
3
  import re
4
  import nltk
5
  import spacy
@@ -7,13 +5,20 @@ import fitz # PyMuPDF
7
  from transformers import pipeline
8
  import textwrap
9
  import gradio as gr
 
 
 
10
 
11
  # Download NLTK punkt if not already done
12
  nltk.download('punkt')
13
 
14
  # Load spaCy model
15
- nlp = spacy.load("en_core_web_sm")
16
-
 
 
 
 
17
  # Initialize the BigBird-Pegasus summarization pipeline for PubMed texts
18
  summarizer = pipeline("summarization", model="google/bigbird-pegasus-large-pubmed")
19
 
 
 
 
1
  import re
2
  import nltk
3
  import spacy
 
5
  from transformers import pipeline
6
  import textwrap
7
  import gradio as gr
8
+ import spacy
9
+
10
+
11
 
12
  # Download NLTK punkt if not already done
13
  nltk.download('punkt')
14
 
15
  # Load spaCy model
16
+ try:
17
+ nlp = spacy.load("en_core_web_sm")
18
+ except Exception as e:
19
+ spacy.cli.download("en_core_web_sm")
20
+ nlp = spacy.load("en_core_web_sm")
21
+
22
  # Initialize the BigBird-Pegasus summarization pipeline for PubMed texts
23
  summarizer = pipeline("summarization", model="google/bigbird-pegasus-large-pubmed")
24