Fill mask modern bert

app.py

@@ -13,7 +13,7 @@ pipe = pipeline(
 
 # Function to run the model
 def fill_mask(input_text):
-    return pipe(input_text)[0]["0"]["sequence"]
+    return pipe(input_text)[0]["sequence"]
 
 
 # Set up the Gradio interface