Update app.py

app.py

@@ -24,7 +24,7 @@ def show_preds(input_image, display_label, display_bbox, detection_threshold):
 
     img = PIL.Image.fromarray(input_image, 'RGB')
 
-    pred_dict  = models.torchvision.faster_rcnn.end2end_detect(img, valid_tfms, model1, class_map=class_map, detection_threshold=detection_threshold,
+    pred_dict  = models.torchvision.faster_rcnn.end2end_detect(img, valid_tfms, model1, class_map=ClassMap(class_map), detection_threshold=detection_threshold,
                                            display_label=display_label, display_bbox=display_bbox, return_img=True, 
                                            font_size=16, label_color="#FF59D6")