Update app.py

app.py

@@ -24,9 +24,15 @@ def find_dataset_path(root_dir):
             return root
     return None
 
-dataset_path = find_dataset_path("data")
+# Look for 'safe' and 'unsafe' inside 'data/Data'
+dataset_path = find_dataset_path("data/Data")
 if dataset_path is None:
-    raise FileNotFoundError("Expected 'safe' and 'unsafe' folders not found after extraction. Please check the Data.zip structure.")
+    print("Debugging extracted structure:")
+    for root, dirs, files in os.walk("data"):
+        print(f"Root: {root}")
+        print(f"Directories: {dirs}")
+        print(f"Files: {files}")
+    raise FileNotFoundError("Expected 'safe' and 'unsafe' folders not found inside 'data/Data'. Please check the Data.zip structure.")
 print(f"Dataset path found: {dataset_path}")
 
 # Step 3: Define Custom Dataset Class
@@ -151,3 +157,4 @@ if __name__ == "__main__":
 
 
 
+