IS361Group4 commited on
Commit
7cc3ec1
·
verified ·
1 Parent(s): ada270a

Update modules/financial_analyst.py

Browse files
Files changed (1) hide show
  1. modules/financial_analyst.py +6 -0
modules/financial_analyst.py CHANGED
@@ -3,6 +3,12 @@ os.system("pip install gradio==3.0.18")
3
  from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification
4
  import gradio as gr
5
  import spacy
 
 
 
 
 
 
6
  nlp = spacy.load('en_core_web_sm')
7
  nlp.add_pipe('sentencizer')
8
 
 
3
  from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification, AutoModelForTokenClassification
4
  import gradio as gr
5
  import spacy
6
+ try:
7
+ nlp = spacy.load("en_core_web_sm")
8
+ except OSError:
9
+ from spacy.cli import download
10
+ download("en_core_web_sm")
11
+ nlp = spacy.load("en_core_web_sm")
12
  nlp = spacy.load('en_core_web_sm')
13
  nlp.add_pipe('sentencizer')
14