Update app.py

app.py

@@ -269,6 +269,8 @@ def remove_overlapping_and_inside_boxes(boxes, classes):
 
 def process_r1(r1):
     # Step 1: Find the index of the maximum value
+    if len(r1) == 2:
+        return r1
     max_index = r1.index(max(r1))
     one_index = r1.index(1)
     # Step 2: Swap the maximum value with 1