Update app.py

app.py

@@ -6,6 +6,8 @@ from scripts.torch_ner_pipe import make_torch_entity_recognizer
 
 
 def main(models: str = None, default_text: str = None):
+    st.title('NER Predictor')
+    st.header('Enter the characteristics of the diamond:')
     models = "training_trf/model-best"
     test = "The patient had surgery."
     models = [name.strip() for name in models.split(",")]
@@ -13,8 +15,6 @@ 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__":