Fill mask modern bert

app.py

@@ -10,19 +10,20 @@ pipe = pipeline(
     torch_dtype=torch.bfloat16,
 )
 
+
 # Function to run the model
 def fill_mask(input_text):
-    return pipe(input_text)
+    return pipe(input_text)[0]["0"]["sequence"]
+
 
 # Set up the Gradio interface
 iface = gr.Interface(
-    fn=fill_mask, 
-    inputs="text", 
+    fn=fill_mask,
+    inputs="text",
     outputs="json",
     title="Mask Filling with ModernBERT",
-    description="Enter a sentence with a [MASK] token, and the model will predict the missing word."
+    description="Enter a sentence with a [MASK] token, and the model will predict the missing word.",
 )
 
 # Launch the interface
 iface.launch()
-