Update app.py

app.py

@@ -50,7 +50,7 @@ railway_file_path = "assets/railway.jpg"
 happy_file_path = "assets/happy.jpg"
 
 unrelated_features = [
-    693, 793, 2033, 1195, 4879, 890, 1762, 45, 4258
+    693, 793, 2033, 1195, 1257, 4879, 890, 1762, 45, 4258
 ]
 
 @spaces.GPU
@@ -83,7 +83,7 @@ def generate_activations(image):
         result.scatter_(-1, topk.indices, topk.values)
         cached_list.append(result.detach().cpu())
         topk_indices = (
-            latents.squeeze(0).mean(dim=0).topk(k=500).indices.detach().cpu()
+            latents.squeeze(0).mean(dim=0).topk(k=1000).indices.detach().cpu()
         )
 
     handles = [hooked_module.register_forward_hook(hook)]
@@ -209,7 +209,7 @@ with gr.Blocks() as demo:
         """
         # Large Multi-modal Models Can Interpret Features in Large Multi-modal Models
 
-        🔍 [ArXiv Paper](https://arxiv.org/abs/2411.14982) | 🏠 [LMMs-Lab Homepage](https://lmms-lab.framer.ai) | 🤗 [Huggingface Collections](https://huggingface.co/collections/lmms-lab/llava-sae-674026e4e7bc8c29c70bc3a3) | [GitHub Repo](https://github.com/EvolvingLMMs-Lab/multimodal-sae) | [Feature Interpretation](https://huggingface.co/datasets/lmms-lab/llava-sae-explanations-5k)
+        🔍 [A Database for Interpreted 5k Neurons](https://huggingface.co/datasets/lmms-lab/llava-sae-explanations-5k) | 🔍 [ArXiv Paper](https://arxiv.org/abs/2411.14982) [ArXiv Paper](https://arxiv.org/abs/2411.14982) | 🏠 [LMMs-Lab Homepage](https://lmms-lab.framer.ai) | 🤗 [Huggingface Collections](https://huggingface.co/collections/lmms-lab/llava-sae-674026e4e7bc8c29c70bc3a3) | [GitHub Repo](https://github.com/EvolvingLMMs-Lab/multimodal-sae) | [Feature Interpretation](https://huggingface.co/datasets/lmms-lab/llava-sae-explanations-5k)
         """
     )
     with gr.Accordion("ℹ️ Instructions", open=False):
@@ -237,10 +237,12 @@ with gr.Blocks() as demo:
                 [
                     ["assets/sunglasses.jpg", 10, "Sunglasses"],
                     ["assets/greedy.jpg", 14, "Greedy eating"],
+                    ["assets/greedy.jpg", 1085, "Human Faces"],
                     ["assets/railway.jpg", 28, "Railway tracks"],
                     ["assets/bird.png", 1803, "The seagull feathers."],
                     ["assets/eyes.png", 2274, "Eyes"],
-                    ["assets/monkey.png", 2692, "Monkey"],
+                    ["assets/happy.jpg", 19379, "Happy"],
+                    ["assets/sad.jpg", 108692, "Sad"],
                 ],
                 inputs=[image, feature_num, dummy_text],
                 label="Examples",