Update app.py

app.py

@@ -156,7 +156,9 @@ if submit and len(x) != 0:
   
   #lstSentEnc,lstSentEnt,lstSentbilbl
   
-  st.text("Entities detected, Next: Relation detection ...")
+  st.text("Entities detected.")
+  st.text("")
+  st.text("Next: Relation detection ...")
   
   
   # Relation extraction part
@@ -219,7 +221,7 @@ if submit and len(x) != 0:
   
   for i in range(len(outrelbl)):
     if outrelbl[i] != "other":
-      edccan += [[lstSentEnc[i],lstSentEnt[i][0], lstSentEnt[i][1],lstSentbilbl[i][0]+" "+outrelbl[i][:-7]+" "+lstSentbilbl[i][1]]]
+      edccan += [[lstSentEnc[i],lstSentEnt[i][0]+" ["+lstSentbilbl[i][0][2:]+"]", lstSentEnt[i][1]+" ["+lstSentbilbl[i][0][2:]+"]",outrelbl[i][:-7]]]
    
   edccandf = pd.DataFrame(edccan, columns= ["Sentence", "Entity 1", "Entity 2", "Relation"] )