Update app.py

app.py

@@ -56,20 +56,20 @@ def visualize_prediction(img, output_dict, threshold=0.5, id2label=None):
     ax = plt.gca()
     colors = COLORS * 100
     for score, (xmin, ymin, xmax, ymax), label, color in zip(scores, boxes, labels, colors):
-    if label == 'license-plates':
+        if label == 'license-plates':
 
-        # Crop plate
-        plate_crop = img.crop((xmin, ymin, xmax, ymax))
+            # Crop plate
+            plate_crop = img.crop((xmin, ymin, xmax, ymax))
 
-        # Check EV
-        ev = is_green_plate(plate_crop)
+            # Check EV
+            ev = is_green_plate(plate_crop)
 
-        if ev:
-            plate_type = "EV (Green Plate)"
-            box_color = "green"
-        else:
-            plate_type = "Non-EV Plate"
-            box_color = "red"
+            if ev:
+                plate_type = "EV (Green Plate)"
+                box_color = "green"
+            else:
+                plate_type = "Non-EV Plate"
+                box_color = "red"
 
         ax.add_patch(
             plt.Rectangle((xmin, ymin), xmax - xmin, ymax - ymin,