test: fixed typo

app.py

@@ -28,7 +28,9 @@ _CACHE = {
     'sae_data_dict': {},
     'model_data': {},
     'segmasks': {},
-    'top_images': {}
+    'top_images': {},
+    'precomputed_activations' = {}
+
 }
 
 def load_all_data(image_root: str, pkl_root: str) -> Tuple[Dict, Dict]:
@@ -638,6 +640,8 @@ def precompute_segmasks():
                 except Exception as e:
                     print(f"Error precomputing mask {cache_key}: {e}")
 
+data_dict, sae_data_dict = load_all_data(image_root="./data/image", pkl_root=pkl_root)
+default_image_name = "christmas-imagenet"
 
 with gr.Blocks(
     theme=gr.themes.Citrus(),
@@ -837,6 +841,13 @@ if __name__ == "__main__":
     
     # Configure cache sizes based on available RAM
     cache_size = int(total_ram_gb * 100)  # Rough estimate: 100 entries per GB
+
+    # Precompute all data
+    print("Starting precomputation...")
+    preload_all_model_data()
+    precompute_activations()
+    precompute_segmasks()
+    print("Precomputation complete!")
     
     # Memory monitoring function
     def monitor_memory_usage():
@@ -859,18 +870,6 @@ if __name__ == "__main__":
     import threading
     start_memory_monitor()
 
-    
-    # Add to initialization
-    preload_all_model_data()
-    
-    _CACHE['precomputed_activations'] = {}
-    precompute_activations()
-    precompute_segmasks()
-    
-    data_dict, sae_data_dict = load_all_data(image_root="./data/image", pkl_root=pkl_root)
-    default_image_name = "christmas-imagenet"
-
-
     # Launch the app with memory-optimized settings
     demo.queue(max_size=min(20, int(total_ram_gb)))  # Scale queue with RAM
     demo.launch(