api access

app.py

@@ -138,6 +138,12 @@ def render_random_examples():
 
     # Display the table
     st.table(st.session_state['random_examples'])
+def predict_using_trained(sentence):
+    model = load_trained_model()
+
+    entities = model(sentence)
+
+    return entities
 
 def prep_page():
     model = load_trained_model()
@@ -198,5 +204,12 @@ def prep_page():
 
 
 if __name__ == '__main__':
-    
-    prep_page()
+
+    query_params = st.query_params
+    if 'api' in query_params:
+        sentence = query_params.get('sentence')
+        entities = predict_using_trained(sentence)
+        st.write({"sentence" : sentence , "entities" : entities})
+    else:
+        prep_page()
+