final changes

app.py

@@ -55,6 +55,7 @@ trained_model, tokenizer, test, NER = load_model()
 st.header("Try out a text!")
 with st.form('model_prediction'):
     text = st.text_area("Input Text", test.iloc[0]['text'][20])
+    text = text[:2000]
     n1, n2, n3 = st.columns((0.2,0.4,0.4))
     ner_check = n1.checkbox("Extract Location and Date", value=True)
     predict = n2.form_submit_button("Predict")
@@ -91,8 +92,8 @@ with st.form('model_prediction'):
                         loc.append(word.text)
                 loc = list(set(loc))
                 date = list(set(date))
-                loc = "None found" if len(loc)==0 else loc
-                date = "None found" if len(date)==0 else date
+                loc = ["None found"] if len(loc)==0 else loc
+                date = ["None found"] if len(date)==0 else date
                 st.write("Location entities: " + ",".join(loc))
                 st.write("Date entities: " + ",".join(date))