Update app.py

app.py

@@ -1,4 +1,4 @@
-import os
+
 try:
     import detectron2
 except:
@@ -23,8 +23,10 @@ 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()
-cfg.MODEL.DEVICE='cpu'
+
 
 # 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")