Update app.py

app.py

@@ -75,6 +75,9 @@ def classifyCar(im):
     outputs = predictor(im)
     v = Visualizer(im[:, :, ::-1], MetadataCatalog.get(cfg.DATASETS.TRAIN[0]), scale=1.2)
     out = v.draw_instance_predictions(outputs["instances"])
+    car_class_true = outputs["instances"].pred_classes == 2
+    boxes = outputs["instances"].pred_boxes[car_class_true]
+    im2 = Image.fromarray(np.uint8(out.get_image())).convert('RGB').crop(list(outputs["instances"].pred_boxes)[1].numpy())
     label = "success"
   except:
     label = "fail2"