Fix: en_core_web_sm

app.py

@@ -6,6 +6,8 @@ import pandas as pd
 import torch
 import math
 import re
+from multiprocessing import Process
+import os
 
 def set_seed(seed):
     torch.manual_seed(seed)
@@ -13,6 +15,9 @@ def set_seed(seed):
         torch.cuda.manual_seed_all(seed)
 set_seed(1212)
 
+def loadEnModel():
+    os.system("python3 -m spacy download en_core_web_sm")
+
 class GramformerDemo:
 
     def __init__(self):
@@ -117,6 +122,7 @@ class GramformerDemo:
             st.stop()
 
         with st.spinner('Loading model..'):
+            loadEnModel()
             gf = self.load_gf(self.model_map[model_type])
     
         input_text = st.selectbox(

gramformer/__init__.py

@@ -1,4 +1 @@
-import os
-os.system("python3 -m spacy download en_core_web_sm")
-
 from gramformer.gramformer import Gramformer