Update app.py

app.py

@@ -2,7 +2,7 @@ from huggingface_hub import from_pretrained_fastai
 import gradio as gr
 from fastai.vision.all import *
 
-class_map = ClassMap(['kangaroo'])
+class_map = {'background': 0, 'kangaroo': 1}
 model = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet50_fpn,
                                              num_classes=2)
 state_dict = torch.load('fasterRCNNCanguros/fasterRCNNCanguros.pth')