init

app.py

@@ -1,35 +1,30 @@
 import gradio as gr
+from transformers import pipeline
 
-title = "BERT"
+# Load the SteelBERT model for mask filling
+fill_mask = pipeline("fill-mask", model="MGE-LLMs/SteelBERT")
 
-description = "Gradio Demo for BERT. To use it, simply add your text, or click one of the examples to load them. Read more at the links below."
+# Define the Gradio interface
+def mask_filler(text):
+    result = fill_mask(text)
+    return result[0]["sequence"]
 
-article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1810.04805' target='_blank'>BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding</a></p>"
+iface = gr.Interface(
+    fn=mask_filler,
+    inputs=gr.Textbox(lines=3, label="Enter a sentence with [MASK] token"),
+    outputs=gr.Textbox(label="Completed sentence"),
+    title="SteelBERT Mask Filler",
+    description="Fill in the masked token using SteelBERT.",
+)
 
+# Add some example inputs for the interface
 examples = [
-    ['Paris is the [MASK] of France.', 'bert-base-cased']
+    ["SteelBERT is a [MASK] model for filling in missing information."],
+    ["I have a [MASK] of apples and oranges."],
 ]
 
-io1 = gr.Interface.load("huggingface/bert-base-cased")
-io2 = gr.Interface.load("huggingface/bert-base-uncased")
+iface.examples = examples
 
-def inference(inputtext, model):
-    if model == "bert-base-cased":
-        outlabel = io1(inputtext)
-    else:
-        outlabel = io2(inputtext)
-    return outlabel
-
-inputs = gr.Textbox(label="Context", lines=10)
-model_choice = gr.Dropdown(choices=["bert-base-cased", "bert-base-uncased"], label="Model")
-outputs = gr.Textbox(label="Output")
-
-gr.Interface(
-    fn=inference,
-    inputs=[inputs, model_choice],
-    outputs=outputs,
-    examples=examples,
-    article=article,
-    title=title,
-    description=description
-).launch(enable_queue=True)
+# Launch the interface
+if __name__ == "__main__":
+    iface.launch()

config.json

@@ -0,0 +1,5 @@
+{
+    "environments": ["python"],
+    "command": "python app.py"
+  }
+