Update app.py

app.py

@@ -129,10 +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 = ''
-    elem_classes = [['bubble', 'even_bubble' if i % 2 == 0 else 'odd_bubble'] + (['faded'] if True else []) for i in range(len(generation_texts))]
+    elem_classes = [['bubble', 'even_bubble' if i % 2 == 0 else 'odd_bubble'] + (['faded_bubble'] 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]
+                      for i, text in enumerate(generation_texts)]
     bubble_outputs += [gr.Textbox('', visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
     return [progress_dummy_output, *bubble_outputs]