Update app.py

app.py

@@ -30,16 +30,19 @@ def add_dpoc_entry(title, description):
     st.session_state.dpoc_list.append(entry)
 # Defect categories
 class_labels = [
+    "Construction Joint",
     "Floor Cracks",
     "Floor Debonding",
-    "Shade Fading",
-    "Air Bubble (Single Defect)",
-    "Spike Roller (Single Defect)"
+    "Floor peeling",
+    "Floor oil",
+    "Flooring Bubble",
+    "Pot holes",
+    "Shade fading"
 ]
 
 @st.cache_resource
 def load_trained_model():
-    return load_model('my_inceptionmodelwithoutaug (1).h5')
+    return load_model('my_inceptionmodelwithoutaug8.h5')
 
 def compress_image(image_bytes, max_size_kb=500):
     img = Image.open(io.BytesIO(image_bytes))