init

app.py

@@ -15,7 +15,7 @@ examples = [
 ]
 
 # Load interfaces for different models
-io1 = gr.Interface.load("huggingface/SteelBERT")
+io1 = gr.Interface.load("huggingface/MGE-LLMs/SteelBERT")
 io2 = gr.Interface.load("huggingface/bert-base-uncased")
 
 def inference(inputtext, model):
@@ -29,9 +29,9 @@ gr.Interface(
     fn=inference, 
     inputs=[
         gr.Textbox(label="Context", lines=10), 
-        gr.Dropdown(choices=["SteelBERT", "bert-base-uncased"], type="value", default="SteelBERT", label="Model")
+        gr.Dropdown(choices=["SteelBERT", "bert-base-uncased"], value="SteelBERT", label="Model")
     ], 
-    outputs=gr.Label(label="Output"),
+    outputs=gr.Textbox(label="Output"),
     examples=examples,
     title=title,
     description=description