Update app.py

app.py

@@ -1,4 +1,5 @@
 import spacy_streamlit
+import streamlit as st
 import typer
 from scripts.torch_ner_model import build_torch_ner_model
 from scripts.torch_ner_pipe import make_torch_entity_recognizer
@@ -12,6 +13,8 @@ def main(models: str = None, default_text: str = None):
     spacy_streamlit.visualize(
         models, default_text, visualizers=["ner"], ner_labels=labels
     )
+    st.title('NER Predictor')
+    st.header('Enter the characteristics of the diamond:')
 
 
 if __name__ == "__main__":