Update app.py

app.py

@@ -129,7 +129,10 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
                                                **generation_kwargs)
     generation_texts = tokenizer.batch_decode(generated)
     progress_dummy_output = ''
-    bubble_outputs = [gr.Textbox(text.replace('\n', ' '), show_label=True, visible=True, container=False, label=f'Layer {i}') for text in generation_texts]
+    elem_classes = [['bubble', 'even_bubble' if i % 2 == 0 else 'odd_bubble'] + (['faded'] if True else []) for i in range(len(generation_texts))]
+    bubble_outputs = [gr.Textbox(text.replace('\n', ' '), show_label=True, visible=True, 
+                                 container=False, label=f'Layer {i}', elem_classes=elem_classes[i])
+                      for text in generation_texts]
     bubble_outputs += [gr.Textbox('', visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
     return [progress_dummy_output, *bubble_outputs]
 
@@ -220,9 +223,8 @@ 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, 
-                                             elem_classes=['bubble', 'even_bubble' if i % 2 == 0 else 'odd_bubble']
-                                            ) for i in range(MAX_NUM_LAYERS)]
+        interpretation_bubbles = [gr.Textbox('', show_label=True, label=f'Layer {i}', container=False, visible=False) 
+                                  for i in range(MAX_NUM_LAYERS)]
         
     # event listeners    
     for i, btn in enumerate(tokens_container):