Update config.json

config.json

@@ -1,7 +1,7 @@
 {
-  "_name_or_path": "nvidia/mit-b0",
+  "_name_or_path": "Xpitfire/segformer-finetuned-segments-cmp-facade",
   "architectures": [
-    "SegformerForSemanticSegmentation"
+    "SegformerForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
   "classifier_dropout_prob": 0.1,
@@ -33,11 +33,11 @@
     "2": "facade", 
     "3": "window", 
     "4": "door", 
-    "5": "pillar", 
+    "5": "cornice", 
     "6": "sill", 
     "7": "balcony", 
     "8": "blind",  
-    "9": "cornice",
+    "9": "pillar",
     "10": "deco", 
     "11": "molding",
     "12": "shop"
@@ -50,11 +50,11 @@
     "facade": 2, 
     "window": 3, 
     "door": 4, 
-    "pillar": 5, 
+    "cornice": 5, 
     "sill": 6, 
     "balcony": 7, 
     "blind": 8,  
-    "cornice": 9,
+    "pillar": 9,
     "deco": 10, 
     "molding": 11,
     "shop": 12