Update app.py

app.py

@@ -103,7 +103,23 @@ tokenizer = AutoTokenizer.from_pretrained(tokenizer_name, token=os.environ['hf_t
 
 # demo
 json_output = gr.JSON()
-with gr.Blocks(theme=gr.themes.Default()) as demo:
+css = '''
+.token_btn{
+  background: none;
+  border: none;
+  padding: 0;
+  font: inherit;
+  cursor: pointer;
+  color: blue; /* default text color */
+  font-weight: bold;
+}
+
+.token_btn:hover {
+    color: red;
+}
+
+'''
+with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
     global_state = gr.State([])
     with gr.Row():
         with gr.Column(scale=5):
@@ -146,7 +162,7 @@ with gr.Blocks(theme=gr.themes.Default()) as demo:
         tokens_container = []
         with gr.Row():
             for i in range(MAX_PROMPT_TOKENS):
-                btn = gr.Button('', visible=False)
+                btn = gr.Button('', visible=False, elem_classes=['token_btn'])
                 btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, num_tokens, do_sample, temperature, 
                                                            top_k, top_p, repetition_penalty, length_penalty
                                                           ], [json_output])