Update app.py

app.py

@@ -114,9 +114,8 @@ css = '''
   border: 2px solid #000;
   border-radius: 10px;
   padding: 10px;
-  margin-left: 20px;
-  margin-right: 20px;
-  margin-top: 10px;
+  margin-left: 5%;
+  width: 90%;
   background: pink;
 }
 .bubble textarea {
@@ -194,10 +193,22 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
             pass
         with gr.Tab('LLM Attacks'):
             pass
+            
     with gr.Group():
         original_prompt_raw = gr.Textbox(value='Should I eat cake or vegetables?', container=True, label='Original Prompt')
         original_prompt_btn = gr.Button('Compute', variant='primary')
-        
+
+    tokens_container = []
+    with gr.Row():
+        for i in range(MAX_PROMPT_TOKENS):
+            btn = gr.Button('', visible=False, elem_classes=['token_btn'])
+            tokens_container.append(btn)
+    
+    progress_dummy = gr.Markdown('', elem_id='progress_dummy')
+
+    with gr.Group('Interpretation'):
+        interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
+
     with gr.Accordion(open=False, label='Settings'):
         with gr.Row():
             num_tokens = gr.Slider(1, 100, step=1, value=20, label='Max. # of Tokens')
@@ -211,26 +222,16 @@ with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
                 top_k = gr.Slider(1, 1000, value=50, step=1, label='top k')
                 top_p = gr.Slider(0., 1., value=0.95, label='top p')
 
-    with gr.Group('Interpretation'):
-        interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')
-
     with gr.Group('Output'):
-        tokens_container = []
-        with gr.Row():
-            for i in range(MAX_PROMPT_TOKENS):
-                btn = gr.Button('', visible=False, elem_classes=['token_btn'])
-                tokens_container.append(btn)
-        
-        progress_dummy = gr.Markdown('', elem_id='progress_dummy')
         interpretation_bubbles = [gr.Textbox('', container=False, visible=False, elem_classes=['bubble'])
                                  for i in range(model.config.num_hidden_layers)]
         
-        for i, btn in enumerate(tokens_container):
-            btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, 
-                                                         num_tokens, do_sample, temperature, 
-                                                         top_k, top_p, repetition_penalty, length_penalty
-                                                        ], [progress_dummy, *interpretation_bubbles])
-        
+    for i, btn in enumerate(tokens_container):
+        btn.click(partial(run_interpretation, i=i), [global_state, interpretation_prompt, 
+                                                     num_tokens, do_sample, temperature, 
+                                                     top_k, top_p, repetition_penalty, length_penalty
+                                                    ], [progress_dummy, *interpretation_bubbles])
+    
     original_prompt_btn.click(get_hidden_states, 
                               [original_prompt_raw], 
                               [progress_dummy, global_state, *tokens_container])