Update app.py

app.py

@@ -27,11 +27,11 @@ import numpy as np
 model_path = "model_final.pth"
 
 cfg = get_cfg()
-cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.6
-cfg.MODEL.ROI_HEADS.NUM_CLASSES = 4
+cfg.MODEL.ROI_HEADS.SCORE_THRESH_TEST = 0.7
+cfg.MODEL.ROI_HEADS.NUM_CLASSES = 2
 cfg.MODEL.WEIGHTS = model_path
 st.write("""
-         # Car Damage Detection
+         # Gun Detection
          """
          )
 file = st.file_uploader("Please upload an image file(JPG/PNG/JPEG format)", type=["jpg", "png","jpeg"])
@@ -39,7 +39,7 @@ file = st.file_uploader("Please upload an image file(JPG/PNG/JPEG format)", type
 st.set_option('deprecation.showfileUploaderEncoding', False)
 
 car_metadata = MetadataCatalog.get("test1")
-car_metadata.thing_classes = ['Damage-car','Damage','Others','Undamage']
+car_metadata.thing_classes = ['guns','Gun',]
 
 if not torch.cuda.is_available():
     cfg.MODEL.DEVICE='cpu'