Update app.py

app.py

@@ -20,9 +20,14 @@ from detectron2.engine import DefaultPredictor
 from detectron2.config import get_cfg
 from detectron2.utils.visualizer import Visualizer
 from detectron2.data import MetadataCatalog
-
-
-model_path = "https://huggingface.co/dbmdz/detectron2-model/resolve/main/model_final.pth"
+from detectron2.data.datasets import register_coco_instances
+try:
+    register_coco_instances("Fiber", {}, "./labels-fiver.json", "/content/Fiber")
+    Fiber_metadata = MetadataCatalog.get("Fiber")
+    dataset_dicts = DatasetCatalog.get("Fiber")
+except:
+    print("there is an issue")
+model_path = "https://huggingface.com/mosidi/detectron2-model-demo/resolve/main/model_final.pth"
 
 cfg = get_cfg()
 cfg.merge_from_file("./configs/detectron2/faster_rcnn_R_50_FPN_3x.yaml")
@@ -30,7 +35,7 @@ cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 cfg.MODEL.WEIGHTS = model_path
 
 my_metadata = MetadataCatalog.get("dbmdz_coco_all")
-my_metadata.thing_classes = ["Illumination", "Illustration"]
+my_metadata.thing_classes = ["Fiber", "Fiber"]
 
 if not torch.cuda.is_available():
     cfg.MODEL.DEVICE = "cpu"