Update app.py

app.py

@@ -136,7 +136,7 @@ def process_cache(cache, saes_dict, model_config, timestep=None):
                 
         averages = torch.mean(sparse_maps, dim=(0, 1))
 
-        top_features = torch.topk(averages, 20).indices
+        top_features = torch.topk(averages, 40).indices
 
         top_features_dict[code] = top_features.cpu().tolist()
         sparse_maps_dict[code] = sparse_maps.cpu().numpy()