Update app.py

app.py

@@ -6,24 +6,24 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/1810.048
 
 examples = [
     ['Paris is the [MASK] of France.', 'bert-base-cased']
-    ['I love [MASK]', 'bert-base-uncased']
 ]
 
 # Lade die Interfaces für die Modelle
-io1 = gr.load("huggingface/bert-base-cased")
-io2 = gr.load("huggingface/bert-base-uncased")
+io1 = gr.Interface.load("huggingface/bert-base-cased")
+io2 = gr.Interface.load("huggingface/bert-base-uncased")
 
 def inference(inputtext, model):
     if "[MASK]" not in inputtext:
-        return "Error: The input text must contain the [MASK] token."
+        return {"error": "The input text must contain the [MASK] token."}
     
     if model == "bert-base-cased":
         outlabel = io1(inputtext)
     elif model == "bert-base-uncased":
         outlabel = io2(inputtext)
     else:
-        outlabel = "Invalid model selected"
-    return str(outlabel)
+        return {"error": "Invalid model selected"}
+
+    return outlabel
 
 # Aktualisierte Gradio-Syntax
 gr.Interface(
@@ -32,7 +32,7 @@ gr.Interface(
         gr.Textbox(label="Context", lines=10, placeholder="Enter text with [MASK] token"),
         gr.Dropdown(choices=["bert-base-cased", "bert-base-uncased"], value="bert-base-cased", label="model")
     ],
-    outputs=gr.Label(label="Output"),
+    outputs=gr.JSON(label="Output"),  # We use JSON to display errors or outputs
     examples=examples,
     article=article,
     title=title,