Update app.py

app.py

@@ -212,10 +212,17 @@ EXAMPLE_TEXTS = [
     "I love eating [MASK] with tomato sauce.",
     "The [MASK] barked loudly at the stranger."
 ]
+with gr.Blocks() as demo:
+
+    # Optional styling (safe even on older Gradio versions)
+    gr.HTML("""
+    <style>
+        #tokens-html {
+            font-family: "JetBrains Mono", monospace;
+        }
+    </style>
+    """)
 
-with gr.Blocks(css="""
-#tokens-html { font-family: 'JetBrains Mono', monospace; }
-""") as demo:
     gr.Markdown(DESCRIPTION)
 
     with gr.Row():
@@ -258,7 +265,7 @@ with gr.Blocks(css="""
         outputs=[tokens_html, pred_out, prob_plot, att_plot, info_box],
     )
 
-    # Also run on change for a smoother experience
+    # Allows instant update without clicking Run
     text_in.change(
         analyze,
         inputs=[text_in, layer_slider],
@@ -270,5 +277,6 @@ with gr.Blocks(css="""
         outputs=[tokens_html, pred_out, prob_plot, att_plot, info_box],
     )
 
+
 if __name__ == "__main__":
     demo.launch()