Update app.py

app.py

@@ -13,7 +13,7 @@ import evaluate
 # ------------------------------
 # 1️⃣ Parameter
 # ------------------------------
-DATA_DIR = "Assets"  
+DATA_DIR = "path_to_dataset"  
 NUM_CLASSES = 3               # z.B. 3 Klassen: Hintergrund, Schaden, Rand
 IMAGE_SIZE = 256              # Bildgröße für Training
 
@@ -38,7 +38,7 @@ mask_transforms = transforms.Compose([
 def preprocess(example):
     example["pixel_values"] = train_transforms(example["image"])
     # Masken als LongTensor für CrossEntropyLoss
-    example["Kiesnest.jpg"] = mask_transforms(example["Kiesnest.jpg"]).long().squeeze(0) 
+    example["labeks"] = mask_transforms(example["label"]).long().squeeze(0) 
     return example
 
 dataset = dataset.map(preprocess)