Upload 6 files

app.py

@@ -1,5 +1,3 @@
-CLASSES = ['Carnavalita', 'Kimo-COno', 'Squizz', 'Oreo', 'Moro', 'Dulce', 'KitKat', 'Cadbury', 'Mega', 'other']
-
 import io
 import json
 import os
@@ -30,12 +28,19 @@ IMAGE_DIR = BASE / "images"
 DATASET_FILE = BASE / "dataset.json"
 MODEL_DIR = BASE / "model"
 GENERATED_DIR = BASE / "generated_dataset"
-CLASSES_FILE = ROOT / "training" / "classes.txt"
+CLASSES_FILE = BASE / "classes.txt"
 
 IMAGE_DIR.mkdir(parents=True, exist_ok=True)
 BASE.mkdir(parents=True, exist_ok=True)
 MODEL_DIR.mkdir(parents=True, exist_ok=True)
 
+DEFAULT_CLASSES = [
+    "Carnavalita", "Kimo-COno", "Squizz", "Oreo", "Moro",
+    "Dulce", "KitKat", "Cadbury", "Mega", "other"
+]
+if not CLASSES_FILE.exists():
+    CLASSES_FILE.write_text("\n".join(DEFAULT_CLASSES) + "\n", encoding="utf-8")
+
 CONFIDENCE_THRESHOLD = float(os.getenv("CONFIDENCE_THRESHOLD", "0.35"))
 MAX_IMAGE_MB = int(os.getenv("MAX_IMAGE_MB", "15"))
 

training/classes.txt

@@ -7,4 +7,4 @@ Dulce
 KitKat
 Cadbury
 Mega
-other
+other