Update app.py

app.py

@@ -37,16 +37,14 @@ def draw_coco_bboxes(img, bboxes, color=(255,255,0), width=5, show=False, export
     return img
 
 
-def traffic_sign_inference(input_img):
+def traffic_sign_inference(img):
     # Build the model from a config file and a checkpoint file
     model = init_detector(config_file, checkpoint_file, device='cpu')
     
     result = inference_detector(model, img)
     
-    # Show the results
-    img = mmcv.imread(img) # numpy -> torch here!
-    img = mmcv.imconvert(img, 'bgr', 'rgb')
-
+    # img = mmcv.imread(img) # numpy -> torch here!
+    # img = mmcv.imconvert(img, 'bgr', 'rgb')
 
     bboxes = result.pred_instances.bboxes
     labels = [SIGNS_CLASSES[l] for l in result.pred_instances.labels]