init

app.py

@@ -1,27 +1,13 @@
 import gradio as gr
-from transformers import pipeline
 
-# Load the SteelBERT model for mask filling
-fill_mask = pipeline("fill-mask", model="MGE-LLMs/SteelBERT")
+def greet(name, intensity):
+    return "Hello, " + name + "!" * intensity
 
-# Define the Gradio interface
-def mask_filler(text):
-    result = fill_mask(text)
-    return result[0]["sequence"]
-
-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.",
+demo = gr.Interface(
+    fn=greet,
+    inputs=["text", gr.Slider(value=2, minimum=1, maximum=10, step=1)],
+    outputs=[gr.Textbox(label="greeting", lines=3)],
 )
 
-# Add some example inputs for the interface
-examples = [
-    ["SteelBERT is a [MASK] model for filling in missing information."],
-    ["I have a [MASK] of apples and oranges."],
-]
-
-iface.examples = examples
-iface.launch()
+if __name__ == "__main__":
+    demo.launch()