Update app.py

app.py

@@ -23,16 +23,15 @@ from PIL import Image
 
 car_metadata = MetadataCatalog.get("my_dataset_val")
 car_metadata.thing_classes = ['Damages', 'Dent', 'Dislocation', 'Scratch', 'Shatter'] 
-if not torch.cuda.is_available():
-    cfg.MODEL.DEVICE='cpu'
 cfg = get_cfg()
 
-
 # add project-specific config (e.g., TensorMask) here if you're not running a model in detectron2's core library
 cfg.merge_from_file("myconfig2.yml")
 cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.5  # set threshold for this model
 
 cfg.MODEL.WEIGHTS = "model_final.pth"
+if not torch.cuda.is_available():
+    cfg.MODEL.DEVICE='cpu'
 predictor = DefaultPredictor(cfg)
 def inference(img):
     im = cv2.imread(img.name)