cogcorp commited on
Commit
e62ff42
·
1 Parent(s): 0e79d38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -2,6 +2,7 @@ import gradio as gr
2
  import pandas as pd
3
  import os
4
  import spacy
 
5
  import numpy as np
6
  import zipfile
7
  import tempfile
@@ -16,7 +17,13 @@ import re
16
  # Put your OpenAI API key here
17
  openai.api_key = os.getenv('OpenAPI')
18
 
19
- nlp = spacy.load('en_core_web_md')
 
 
 
 
 
 
20
 
21
  vendor_df_dict = {}
22
 
 
2
  import pandas as pd
3
  import os
4
  import spacy
5
+ import spacy.cli
6
  import numpy as np
7
  import zipfile
8
  import tempfile
 
17
  # Put your OpenAI API key here
18
  openai.api_key = os.getenv('OpenAPI')
19
 
20
+ try:
21
+ nlp = spacy.load('en_core_web_md')
22
+ except IOError:
23
+ print("Model not found. Downloading...")
24
+ spacy.cli.download("en_core_web_md")
25
+ nlp = spacy.load('en_core_web_md')
26
+
27
 
28
  vendor_df_dict = {}
29