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.median() + diff_score2 / diff_score2.median()
+    diff_score = diff_score1 / diff_score1.max().values + diff_score2 / diff_score2.max().values
+    
     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
     diff_score = diff_score[avoid_first-1:len(diff_score)-avoid_last]