Ankush05 commited on
Commit
0112bc1
·
1 Parent(s): 297882c
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,6 +7,6 @@ fill_masker = pipeline(model="bert-base-uncased")
7
  # fill_masker("This is a simple [MASK].")
8
 
9
  if query := st.chat_input("filler: "):
10
- ans = fill_masker(f"[MASK]{query}[MASK]")
11
  with st.chat_message("User"):
12
- st.write(ans)
 
7
  # fill_masker("This is a simple [MASK].")
8
 
9
  if query := st.chat_input("filler: "):
10
+ ans = fill_masker(f"{query}[MASK]")
11
  with st.chat_message("User"):
12
+ st.write(ans[sequence])