Update app.py

app.py

@@ -152,7 +152,7 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
     avoid_first, avoid_last = 2, 1 # layers that are usually never important
     assert avoid_first >= 1 # due to .diff() we will not be able to compute a score for the first layer
     diff_score = diff_score[avoid_first-1:len(diff_score)-avoid_last]
-    important_idxs = avoid_first + diff_score.topk(k=4).indices.cpu().numpy() # k=int(np.ceil(0.15 * len(generation_texts)))
+    important_idxs = avoid_first + diff_score.topk(k=7).indices.cpu().numpy() # k=int(np.ceil(0.15 * len(generation_texts)))
     
     # create GUI output
     print(f'{important_idxs=}')
@@ -179,7 +179,7 @@ for i in range(MAX_PROMPT_TOKENS):
     btn = gr.Button('', visible=False, elem_classes=['token_btn'])
     tokens_container.append(btn)
 
-with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
+with gr.Blocks(theme=gr.themes.Glass(), css='styles.css') as demo:
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown('# 😎 Self-Interpreting Models')