Update app.py

app.py

@@ -54,11 +54,7 @@ def inference(image):
     #img = np.array(image.resize((800, height)))
     img = np.array(image)
     outputs = predictor(img) 
-    v = Visualizer(img[:, :, ::-1],
-                   metadata=gun_metadata, 
-                   scale=0.5, 
-                   instance_mode=ColorMode.IMAGE_BW  
-    )
+    v=Visualizer(im,MetadataCatalog.get(dataset_name),scale=1)
     out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
     return out.get_image()
 if file is None: