Update app.py

app.py

@@ -18,10 +18,9 @@ target_list = [
     'Hohlräume', 'Abplatzung', 'Graffiti', 'Verwitterung', 
     'Restschalung', 'freiliegendeBewehrung', 'Lager', 'Entwässerung', 'Verfärbung' 
 ]
-target_list_all = ["All"] + target_list 
-classes, nclasses = target_list, len(target_list)
-label2id = dict(zip(classes, range(nclasses)))
-id2label = dict(zip(range(nclasses), classes))
+
+label2id = {c: i for i, c in enumerate(target_list)}
+id2label = {i: c for i, c in laberl2id.items()}
 
 
 # SegModel