Update app.py

app.py

@@ -7,9 +7,19 @@ 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())
+# 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)
+
+ui = gr.Interface(
+    fn=mask,
+    inputs="textbox",
+    outputs="textbox",
+    submit_btn="Predict",
+    title="BERT Test - Zircon.tech",
+    description="Testing BERT Model"
+    )
 ui.launch(share=True)