Update app.py
Browse files
app.py
CHANGED
|
@@ -11,15 +11,16 @@ with gr.Blocks() as demo:
|
|
| 11 |
#tab single model
|
| 12 |
with gr.Tab("Single-Models"):
|
| 13 |
with gr.Row(equal_height=True):
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
|
| 19 |
-
|
| 20 |
inputs = textbox_mask,
|
| 21 |
outputs = textbox_mask_output)
|
| 22 |
demo.launch(debug=True)
|
|
|
|
| 23 |
|
| 24 |
# # Load the model and tokenizer
|
| 25 |
# tokenizer = BertTokenizer.from_pretrained("google-bert/bert-base-uncased")
|
|
|
|
| 11 |
#tab single model
|
| 12 |
with gr.Tab("Single-Models"):
|
| 13 |
with gr.Row(equal_height=True):
|
| 14 |
+
with gr.Column(scale=1):
|
| 15 |
+
textbox_mask = gr.Textbox(label="Give me a sentance and use the word [MASK] to predict what will go there")
|
| 16 |
+
mask_button = gr.Button("Predict")
|
| 17 |
+
textbox_mask_output = gr.Textbox()
|
| 18 |
|
| 19 |
+
mask_button.click(fn=mask,
|
| 20 |
inputs = textbox_mask,
|
| 21 |
outputs = textbox_mask_output)
|
| 22 |
demo.launch(debug=True)
|
| 23 |
+
|
| 24 |
|
| 25 |
# # Load the model and tokenizer
|
| 26 |
# tokenizer = BertTokenizer.from_pretrained("google-bert/bert-base-uncased")
|