Update app.py

app.py

@@ -44,18 +44,21 @@ reversed_map = {
     16: 'Will Smith'
 }
 def extract(zip_file, extract_path):
-    # Create a directory to store extracted images
     os.makedirs(extract_path, exist_ok=True)
-
-    # Extract images from the ZIP file
+    
     with zipfile.ZipFile(zip_file, 'r') as zip_ref:
         zip_ref.extractall(extract_path)
-
-    # Display each image in the extracted directory
-    image_files = os.listdir(extract_path)
-    for image_file in image_files:
-        image_path = os.path.join(extract_path, image_file)
-        image = Image.open(image_path)
+    
+    for root, dirs, files in os.walk(extract_path):
+        for file in files:
+            if file.lower().endswith(('.png', '.jpg', '.jpeg', '.gif')):  # Adjust file extensions as needed
+                image_path = os.path.join(root, file)
+                try:
+                    image = Image.open(image_path)
+                    # Process or display the image here
+                    image.show()
+                except IOError as e:
+                    print(f"Error opening image {image_path}: {e}")
         
 model = timm.create_model("rexnet_150", pretrained = True, num_classes = 17)  
 model.load_state_dict(torch.load('faces_best_model.pth', map_location=torch.device('cpu')))