Upload app.py

app.py

@@ -15,6 +15,11 @@ checkpoints = [
 ]
 
 
+def download_spacy_model(model="en"):
+    spacy.cli.download(model)  # type: ignore
+    return True
+
+
 @st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def get_model(model_name):
     return HappyTextToText("T5", model_name)
@@ -45,6 +50,7 @@ def main():
         """This writing assistant will proofread any text for you! See my [GitHub repo](https://github.com/aseifert/hf-writing-assistant) for implementation details."""
     )
 
+    download_spacy_model()
     annotator = get_annotator("en")
     checkpoint = st.selectbox("Choose model", checkpoints)
     model = get_model(checkpoint)