Update app.py

app.py

@@ -146,7 +146,8 @@ def run_interpretation(raw_original_prompt, raw_interpretation_prompt, max_new_t
                                 -len(bags_of_words[i+1] & bags_of_words[i]) / np.sqrt(len(bags_of_words[i+1]) * len(bags_of_words[i]))
                                 for i in range(len(bags_of_words)-1)
                               ])
-    diff_score = diff_score1 / diff_score1.max() + diff_score2 / diff_score2.max()
+    diff_score = ((diff_score1 - diff_score1.min()) / (diff_score1.max() - diff_score1.min()) 
+                  + (diff_score2 - diff_score2.min()) / (diff_score2.max() - diff_score2.min()))
     
     avoid_first, avoid_last = 2, 1 # layers that are usually never important
     assert avoid_first >= 1 # due to .diff() we will not be able to compute a score for the first layer
@@ -222,7 +223,8 @@ with gr.Blocks(theme=gr.themes.Default(), css='styles.css') as demo:
         with gr.Group():
             raw_original_prompt.render()
             original_prompt_btn = gr.Button('Output Token List', variant='primary')
-    
+            gr.Markdown('**Tokens will appear in the "Tokens" section**')
+            
         gr.Markdown('## Choose Your Interpretation Prompt')
         with gr.Group('Interpretation'):
             raw_interpretation_prompt = gr.Text(suggested_interpretation_prompts[0], label='Interpretation Prompt')