App Bug fix.

app.py

@@ -14,6 +14,7 @@ from transformers import AutoFeatureExtractor, AutoModelForObjectDetection
 from PIL import Image
 import matplotlib.pyplot as plt
 
+id2label = {1: 'person', 2: 'rider', 3: 'car', 4: 'bus', 5: 'truck', 6: 'bike', 7: 'motor', 8: 'traffic light', 9: 'traffic sign', 10: 'train'}
 
 class Detr(pl.LightningModule):
 
@@ -85,8 +86,6 @@ COLORS = [[0.000, 0.447, 0.741], [0.850, 0.325, 0.098], [0.929, 0.694, 0.125],
           [0.184, 0.494, 0.741], [0.494, 0.674, 0.556], [0.494, 0.301, 0.933],
           [0.000, 0.325, 0.850], [0.745, 0.301, 0.188]]
 
-id2label = {1: 'person', 2: 'rider', 3: 'car', 4: 'bus', 5: 'truck', 6: 'bike', 7: 'motor', 8: 'traffic light', 9: 'traffic sign', 10: 'train'}
-
 # for output bounding box post-processing
 def box_cxcywh_to_xyxy(x):
     x_c, y_c, w, h = x.unbind(1)