Update app.py

app.py

@@ -29,24 +29,20 @@ class ImageSearchSystem:
 
         # Prune the model (access vision module correctly)
         parameters_to_prune = (
-            (self.model.vision_model.encoder.layers, 'attention.self.query.weight'),
-            (self.model.vision_model.encoder.layers, 'attention.self.key.weight'),
-            (self.model.vision_model.encoder.layers, 'attention.self.value.weight')
+            (self.model.vision_model.encoder.layers[0].attention.self.query.weight, 'attention.self.query.weight'),
+            (self.model.vision_model.encoder.layers[0].attention.self.key.weight, 'attention.self.key.weight'),
+            (self.model.vision_model.encoder.layers[0].attention.self.value.weight, 'attention.self.value.weight')
         )
 
         # Prune the weights
-        prune.l1_unstructured(parameters_to_prune, amount=0.2)
+        for param, name in parameters_to_prune:
+            prune.l1_unstructured(param, amount=0.2)
         
         # Initialize dataset
         self.image_paths = []
         self.index = None
         self.initialized = False
-
-        # Initialize dataset
-        self.image_paths = []
-        self.index = None
-        self.initialized = False
-        
+    
     def initialize_dataset(self) -> None:
         """Download and process dataset"""
         try: