Update app.py

app.py

@@ -131,6 +131,7 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
     generation_texts = tokenizer.batch_decode(generated)
 
     # create GUI output
+    print(f'{interpreted_vectors.size()=}\t\t{int(np.ceil(0.2 * len(generation_texts)))=}')
     important_idxs = 1 + interpreted_vectors.squeeze().diff(dim=0).topk(k=int(np.ceil(0.2 * len(generation_texts))), dim=0).indices.cpu().numpy()
     progress_dummy_output = ''
     elem_classes = [['bubble', 'even_bubble' if i % 2 == 0 else 'odd_bubble'] + 
@@ -228,7 +229,7 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
                 btn.render()
         
         progress_dummy = gr.Markdown('', elem_id='progress_dummy')
-        interpretation_bubbles = [gr.Textbox('', show_label=True, label=f'Layer {i}', container=False, visible=False) 
+        interpretation_bubbles = [gr.Textbox('', container=False, visible=False) 
                                   for i in range(MAX_NUM_LAYERS)]
         
     # event listeners