hardcode unrelated features

app.py

@@ -49,6 +49,10 @@ greedy_file_path = "assets/greedy.jpg"
 railway_file_path = "assets/railway.jpg"
 happy_file_path = "assets/happy.jpg"
 
+unrelated_features = [
+    693, 793, 2033, 1195, 4879, 890, 1762, 45, 4258
+]
+
 @spaces.GPU
 def generate_activations(image):
     prompt = "<image>"
@@ -96,7 +100,7 @@ def generate_activations(image):
             handle.remove()
     examples = []
     for indice in topk_indices:
-        if indice <= 5000:
+        if indice <= 5000 and indice not in unrelated_features:
             examples.append([ f"model.layers.24_feature{indice.item()}",explanations[f"model.layers.24_feature{indice.item()}"]])