Update app.py

app.py

@@ -9,13 +9,19 @@ def main():
     st.markdown('Mask a word with [MASK]. For example, **I [MASK] dog for a walk**')
     masked = st.text_input('Masked sentence','Please insert a sentence')
     
-    mask_str = '[MASK]'
-    mask_check = re.search(masked, mask_str)
+    submitted = st.form_submit_button("Submit")
     
-    if mask_check:
-        output = unmasker(masked)
-    else:
-        print('Sentence is not masked!!!')
+    if submitted:
+        mask_str = '[MASK]'
+        mask_check = re.search(masked, mask_str)
+        
+        if mask_check:
+            st.success("Submitted")
+            output = unmasker(masked)
+            
+            st.write('Filled sentence:',output[0]['sequence'])
+        else:
+            st.write('Sentence is not masked!!!')
   
 if __name__ == "__main__":
     main()