changes

app.py

@@ -62,7 +62,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", " ".join(test.iloc[0]['text'])[:1525])
+    text = st.text_area("Input Text", test.iloc[0]['text'][20])
     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")
@@ -89,18 +89,20 @@ with st.form('model_prediction'):
             if ner_check:
                 #Perform NER on a single text 
                 n_text = NER(text)
-                loc = ''
-                date = ''
+                loc = []
+                date = []
                 for word in n_text.ents:
                     print(word.text,word.label_)
                     if word.label_ == 'DATE':
-                        date += word.text + ', '
+                        date.append(word.text)
                     elif word.label_ == 'GPE':
-                        loc += word.text + ', '
-                loc = "None found" if len(loc)<1 else loc
-                date = "None found" if len(date)<1 else date
-                st.write("Location entities: " + loc)
-                st.write("Date entities: " + date)
+                        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
+                st.write("Location entities: " + ",".join(loc))
+                st.write("Date entities: " + ",".join(date))
                 
                 #Display entities
                 st.write("All Entities-")