Modify: requirements.txt; Fix: load 'en' in Gramformer

README.md

@@ -4,7 +4,6 @@ emoji: 💩
 colorFrom: blue
 colorTo: gray
 sdk: streamlit
-sdk_version: 0.79.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -146,4 +146,4 @@ class GramformerDemo:
         
 if __name__ == "__main__":
     obj = GramformerDemo()
-    obj.main()
+    obj.main()

gramformer/gramformer.py

@@ -5,7 +5,9 @@ class Gramformer:
     from transformers import AutoModelForSeq2SeqLM
     from lm_scorer.models.auto import AutoLMScorer as LMScorer
     import errant
-    self.annotator = errant.load('en')
+    import en_core_web_sm
+    nlp = en_core_web_sm.load()
+    self.annotator = errant.load('en', nlp)
     
     if use_gpu:
         device= "cuda:0"

requirements.txt

@@ -6,4 +6,5 @@ python-Levenshtein==0.12.2
 fuzzywuzzy==0.18.0
 tokenizers==0.10.2
 fsspec==2021.5.0
-errant
+spacy==2.3.0
+errant==2.2.0