init

app.py

@@ -1,6 +1,33 @@
+# import gradio as gr
+
+# title = "SteelBERT"
+
+# examples = [
+#     ['Paris is the [MASK] of France.', 'SteelBERT'],
+#     ["A composite steel plate for marine construction was fabricated using 316L stainless steel.", 'SteelBERT'],
+#     ["The use of composite [MASK] in construction is growing rapidly.", 'SteelBERT'],
+#     ["Advances in [MASK] science are leading to stronger and more durable steel products.", 'SteelBERT'],
+#     ["The corrosion resistance of stainless steel is attributed to the [MASK] of a passive film on the surface.", 'SteelBERT'],
+#     ["Heat treatment of steel involves a controlled [MASK] and cooling process to alter its mechanical properties.", 'SteelBERT'],
+#     ["Nano-engineered [MASK] have the potential to revolutionize the steel industry with their superior properties.", 'SteelBERT']
+# ]
+
+# # Load interfaces for different models
+# try:
+#     io1 = gr.Interface.load("MGE-LLMs/SteelBERT")
+
+
 import gradio as gr
 
-title = "SteelBERT"
+title = "BERT"
+
+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."
+
+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>"
+
+# examples = [
+#     ['Paris is the [MASK] of France.','bert-base-uncased']
+# ]
 
 examples = [
     ['Paris is the [MASK] of France.', 'SteelBERT'],
@@ -12,36 +39,25 @@ examples = [
     ["Nano-engineered [MASK] have the potential to revolutionize the steel industry with their superior properties.", 'SteelBERT']
 ]
 
-# Load interfaces for different models
-try:
-    io1 = gr.Interface.load("MGE-LLMs/SteelBERT")
-except Exception as e:
-    print(f"Failed to load SteelBERT: {e}")
-    io1 = None
 
-try:
-    io2 = gr.Interface.load("huggingface/bert-base-uncased")
-except Exception as e:
-    print(f"Failed to load bert-base-uncased: {e}")
-    io2 = None
+io1 = gr.Interface.load("MGE-LLMs/SteelBERT")
 
-def inference(inputtext, model):
-    if model == "SteelBERT" and io1 is not None:
-        return io1(inputtext)
-    elif model == "bert-base-uncased" and io2 is not None:
-        return io2(inputtext)
-    else:
-        return "Model not available."
-
-with gr.Blocks(title=title) as demo:
-    gr.Markdown(f"# {title}")
-
-    context = gr.Textbox(label="Context", lines=10)
-    model = gr.Dropdown(choices=["SteelBERT", "bert-base-uncased"], label="Model")
-    output = gr.Textbox(label="Output")
+io2 = gr.Interface.load("huggingface/bert-base-uncased")
 
-    examples_component = gr.Examples(examples=examples, inputs=[context, model])
 
-    btn = gr.Button("Run", fn=inference, inputs=[context, model], outputs=output)
-
-demo.launch()
+def inference(inputtext, model):
+    if model == "SteelBERT":
+        outlabel = io1(inputtext)
+    else:
+        outlabel = io2(inputtext)
+    return outlabel
+     
+
+gr.Interface(
+    inference, 
+    [gr.inputs.Textbox(label="Context",lines=10),gr.inputs.Dropdown(choices=["SteelBERT","bert-base-uncased"], type="value", default="SteelBERT", label="model")], 
+    [gr.outputs.Label(label="Output")],
+    examples=examples,
+    article=article,
+    title=title,
+    description=description).launch(enable_queue=True)