init

app.py

@@ -15,24 +15,37 @@ examples = [
 ]
 
 # Load interfaces for different models
-io1 = gr.Interface.load("huggingface/MGE-LLMs/SteelBERT")
-io2 = gr.Interface.load("huggingface/bert-base-uncased")
+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
 
 def inference(inputtext, model):
-    if model == "SteelBERT":
-        outlabel = io1(inputtext)
+    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:
-        outlabel = io2(inputtext)
-    return outlabel
-
-gr.Interface(
-    fn=inference, 
-    inputs=[
-        gr.Textbox(label="Context", lines=10), 
-        gr.Dropdown(choices=["SteelBERT", "bert-base-uncased"], value="SteelBERT", label="Model")
-    ], 
-    outputs=gr.Textbox(label="Output"),
-    examples=examples,
-    title=title,
-    description=description
-).launch(enable_queue=True)
+        return "Model not available."
+
+with gr.Blocks() as demo:
+    gr.Markdown(f"# {title}")
+    gr.Markdown(description)
+
+    context = gr.Textbox(label="Context", lines=10)
+    model = gr.Dropdown(choices=["SteelBERT", "bert-base-uncased"], value="SteelBERT", label="Model")
+    output = gr.Textbox(label="Output")
+
+    examples_component = gr.Examples(examples=examples, inputs=[context, model])
+
+    btn = gr.Button("Run")
+    btn.click(fn=inference, inputs=[context, model], outputs=output)
+
+demo.launch(enable_queue=True)