Update app.py

app.py

@@ -128,7 +128,7 @@ async def polygon_endpoint(file: UploadFile = File(...)):
     cls_id = int(instances.pred_classes[idx].item())
 
     # Choose class name
-    class_names = {0: "rectangular", 1: "irregular"}
+    class_names = {0: "irregular", 1: "rectangular"}
     chosen_class = class_names.get(cls_id, "unknown")
 
     # --- Choose postprocessing dynamically ---