Update app.py

app.py

@@ -86,20 +86,24 @@ try:
 
 except Exception as e:
     print(f"Error loading or processing dataset: {str(e)}")
-    print("Attempting to load from local path...")
-    local_path = "/home/user/app/uruguayai/montevideo"
-    if os.path.exists(local_path):
-        print(f"Local path exists. Contents: {os.listdir(local_path)}")
-        dataset = load_dataset("imagefolder", data_dir=local_path)
-        print("Dataset structure:", dataset)
-        print("Available splits:", dataset.keys())
-        if "train" in dataset:
-            processed_dataset = dataset["train"].map(preprocess_images, batched=True, remove_columns=dataset["train"].column_names)
-            print(f"Processed dataset size: {len(processed_dataset)}")
-        else:
-            raise ValueError("The local dataset does not contain a 'train' split.")
-    else:
-        raise ValueError(f"Local path {local_path} does not exist.")
+    print("Attempting to find dataset...")
+    
+    # List contents of current directory and parent directories
+    print("Current directory contents:")
+    print(os.listdir('.'))
+    print("Parent directory contents:")
+    print(os.listdir('..'))
+    print("Root directory contents:")
+    print(os.listdir('/'))
+    
+    # Try to find any directory that might contain the dataset
+    for root, dirs, files in os.walk('/'):
+        if 'montevideo' in dirs:
+            print(f"Found 'montevideo' directory at: {os.path.join(root, 'montevideo')}")
+            print(f"Contents: {os.listdir(os.path.join(root, 'montevideo'))}")
+    
+    raise ValueError("Unable to locate or load the dataset. Please check the dataset path and permissions.")
+
 
 # Training function
 def train_step(state, batch, rng):