Update app.py

app.py

@@ -4,7 +4,19 @@ import json
 import requests
 import time
 import os
-  
+
+def load_models():
+  if not is_port_in_use(8080):
+    with st.spinner(text="The model is loading."):
+      proc = Process(target=start_server, args=(), daemon=True)
+      proc.start()
+      while not is_port_in_use(8080):
+        time.sleep(1)
+      st.success("Model server started.")
+  else:
+    st.success("The model has loaded.")
+  st.session_state['models_loaded'] = True
+
 def get_correction(input_text):
   st.markdown(f'##### Corrected text:')
   st.write('')