dar-tau commited on
Commit
518abab
·
verified ·
1 Parent(s): 3691d69

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -84,7 +84,9 @@ def run_interpretation(global_state, raw_interpretation_prompt, max_new_tokens,
84
  generated = interpretation_prompt.generate(model, {0: interpreted_vectors}, k=3, **generation_kwargs)
85
  generation_texts = tokenizer.batch_decode(generated)
86
  progress_dummy_output = ''
87
- return [progress_dummy_output] + [gr.Textbox(text, visible=True, container=False) for text in generation_texts]
 
 
88
 
89
 
90
  ## main
 
84
  generated = interpretation_prompt.generate(model, {0: interpreted_vectors}, k=3, **generation_kwargs)
85
  generation_texts = tokenizer.batch_decode(generated)
86
  progress_dummy_output = ''
87
+ return ([progress_dummy_output] +
88
+ [gr.Textbox(text.replace('\n', ' '), visible=True, container=False) for text in generation_texts]
89
+ )
90
 
91
 
92
  ## main