Initial commit

app.py

@@ -14,7 +14,8 @@ import time
 import os
 
 def start_server():   
-    os.system("python -m spacy download en_core_web_sm")     
+    os.system("python -m spacy download en_core_web_sm")    
+    spacy.load('en_core_web_sm') 
     os.system("uvicorn InferenceServer:app --port 8080 --host 0.0.0.0 --workers 1")
 
 def load_models():
@@ -161,43 +162,43 @@ def get_correction(input_text):
           
         
 if __name__ == "__main__":
-    if not st.session_state['models_loaded']:
-        load_models() 
-
     
-    st.title('Gramformer')
-    st.subheader('A framework for correcting english grammatical errors')
-    st.markdown("Built for fun with 💙  by Prithivi Da, The maker of [WhatTheFood](https://huggingface.co/spaces/prithivida/WhatTheFood), [Styleformer](https://github.com/PrithivirajDamodaran/Styleformer) and [Parrot paraphraser](https://github.com/PrithivirajDamodaran/Parrot_Paraphraser) | ✍️ [@prithivida](https://twitter.com/prithivida) |[[GitHub]](https://github.com/PrithivirajDamodaran)", unsafe_allow_html=True)
-
-    examples = [
-                "what be the reason for everyone leave the comapny",
-                "The girls's backpacks are gone in the morning",
-                "I am doing fine. How is you?",
-                "Each of you all should run fast", 
-                "Matt like fish",
-                "the collection of letters was original used by the ancient Romans",
-                "We enjoys horror movies",
-                "Anna and Mike is going skiing",
-                "I walk to the store and I bought milk",
-                " We all eat the fish and then made dessert",
-                "I will eat fish for dinner and drank milk",
-                ]
-
-    nlp = spacy.load('en_core_web_sm')
-    annotator = errant.load('en', nlp)
-
-    input_text = st.selectbox(
-    label="Choose an example",
-    options=examples
-    )
-    st.write("(or)")
-    input_text = st.text_input(
-        label="Bring your own sentence",
-        value=input_text
-    )
-
-    if input_text.strip(): 
-        get_correction(input_text)
+        st.title('Gramformer')
+        st.subheader('A framework for correcting english grammatical errors')
+        st.markdown("Built for fun with 💙  by Prithivi Da, The maker of [WhatTheFood](https://huggingface.co/spaces/prithivida/WhatTheFood), [Styleformer](https://github.com/PrithivirajDamodaran/Styleformer) and [Parrot paraphraser](https://github.com/PrithivirajDamodaran/Parrot_Paraphraser) | ✍️ [@prithivida](https://twitter.com/prithivida) |[[GitHub]](https://github.com/PrithivirajDamodaran)", unsafe_allow_html=True)
+
+        examples = [
+                    "what be the reason for everyone leave the comapny",
+                    "The girls's backpacks are gone in the morning",
+                    "I am doing fine. How is you?",
+                    "Each of you all should run fast", 
+                    "Matt like fish",
+                    "the collection of letters was original used by the ancient Romans",
+                    "We enjoys horror movies",
+                    "Anna and Mike is going skiing",
+                    "I walk to the store and I bought milk",
+                    " We all eat the fish and then made dessert",
+                    "I will eat fish for dinner and drank milk",
+                    ]
+
+        if not st.session_state['models_loaded']:
+            load_models()                     
+
+        nlp = spacy.load('en_core_web_sm')
+        annotator = errant.load('en', nlp)
+
+        input_text = st.selectbox(
+        label="Choose an example",
+        options=examples
+        )
+        st.write("(or)")
+        input_text = st.text_input(
+            label="Bring your own sentence",
+            value=input_text
+        )
+
+        if input_text.strip(): 
+            get_correction(input_text)