Update app.py

app.py

@@ -32,14 +32,19 @@ except:
     print("there is an issue")
 model_path = "./model_final.pth"
 
-cfg = get_cfg()
-cfg.merge_from_file("./configs/detectron2/faster_rcnn_R_50_FPN_3x.yaml")
-cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
-cfg.MODEL.WEIGHTS = model_path
-
-my_metadata = MetadataCatalog.get("dbmdz_coco_all")
-my_metadata.thing_classes = ["Fiber", "Fiber"]
+# cfg = get_cfg()
+# cfg.merge_from_file("./configs/detectron2/faster_rcnn_R_50_FPN_3x.yaml")
+# cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
+# cfg.MODEL.WEIGHTS = model_path
 
+# my_metadata = MetadataCatalog.get("dbmdz_coco_all")
+# my_metadata.thing_classes = ["Fiber", "Fiber"]
+cfg = get_cfg()
+cfg.merge_from_file("./detectron2_repo/configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml")
+cfg.MODEL.WEIGHTS = os.path.join(cfg.OUTPUT_DIR, "model_final.pth")
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.05  # set the testing threshold for this model
+cfg.DATASETS.TEST = ("fiber", )
+# predictor = DefaultPredictor(cfg)
 if not torch.cuda.is_available():
     cfg.MODEL.DEVICE = "cpu"