Ankush05 commited on
Commit
96a224e
·
1 Parent(s): f16efa1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,4 +9,4 @@ fill_masker = pipeline(model="bert-base-uncased")
9
  if query := st.chat_input("filler: "):
10
  ans = fill_masker(f"{query}[MASK]")
11
  with st.chat_message("User"):
12
- st.write(ans[0][sequence])
 
9
  if query := st.chat_input("filler: "):
10
  ans = fill_masker(f"{query}[MASK]")
11
  with st.chat_message("User"):
12
+ st.write(ans)