Update app.py

app.py

@@ -38,8 +38,8 @@ 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 = ['guns','Gun',]
+gun_metadata = MetadataCatalog.get("gun_data")
+gun_metadata.thing_classes = ['guns','Gun']
 
 if not torch.cuda.is_available():
     cfg.MODEL.DEVICE='cpu'
@@ -52,7 +52,7 @@ def inference(image):
     img = np.array(image)
     outputs = predictor(img) 
     v = Visualizer(img[:, :, ::-1],
-                   metadata=car_metadata, 
+                   metadata=gun_metadata, 
                    scale=0.5, 
                    instance_mode=ColorMode.IMAGE_BW  
     )