Update app.py

app.py

@@ -115,10 +115,8 @@ def optimize_route(distances: np.ndarray) -> List[int]:
     
     # Convert assignment to tour
     tour = []
-    current = 0
-    for _ in range(n):
-        tour.append(current)
-        current = col_ind[current]
+    for i in range(n):
+        tour.append(col_ind[i])
     
     return tour