dar-tau commited on
Commit
7c696c7
·
verified ·
1 Parent(s): 5ba44ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
129
  **generation_kwargs)
130
  generation_texts = tokenizer.batch_decode(generated)
131
  progress_dummy_output = ''
132
- bubble_outputs = [gr.Textbox(text.replace('\n', ' '), visible=True, container=False, label=f'Layer {i}') for text in generation_texts]
133
  bubble_outputs += [gr.Textbox('', visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
134
  return [progress_dummy_output, *bubble_outputs]
135
 
 
129
  **generation_kwargs)
130
  generation_texts = tokenizer.batch_decode(generated)
131
  progress_dummy_output = ''
132
+ bubble_outputs = [gr.Textbox(text.replace('\n', ' '), show_label=True, visible=True, container=False, label=f'Layer {i}') for text in generation_texts]
133
  bubble_outputs += [gr.Textbox('', visible=False) for _ in range(MAX_NUM_LAYERS - len(bubble_outputs))]
134
  return [progress_dummy_output, *bubble_outputs]
135