Update app.py

app.py

@@ -7,19 +7,14 @@ def mask(text):
   return output[0]['sequence']
 
 # Gradio UI
-# with gr.Blocks() as ui:
-#   gr.Markdown("BERT Test - Zircon.tech")
-#   gr.Button("Predict").click(fn=mask,
-#   inputs = gr.Textbox(label="Give me a sentance and use the word [MASK] to predict what will go there"),
-#   outputs = gr.Textbox())
-# ui.launch(share=True)
-
+examples=[['Today I went to [MASK] after I got out of bed.']]
 ui = gr.Interface(
     fn=mask,
-    inputs="textbox",
-    outputs="textbox",
+    inputs="text",
+    outputs="text",
     submit_btn="Predict",
-    title="BERT Test - Zircon.tech",
-    description="Testing BERT Model"
+    title="BERT-based Text Analyzer",
+    description="Enter some text to analyze with BERT!",
+    examples=examples
     )
-ui.launch(share=True)
+ui.launch(share=True)