Update app.py

app.py

@@ -6,6 +6,14 @@ import PIL
 
 class_map = ['raccoon']
 
+presize = 512
+size = 384
+
+train_tfms = tfms.A.Adapter(
+    [*tfms.A.aug_tfms(size=size, presize=presize), tfms.A.Normalize()]
+)
+valid_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(size), tfms.A.Normalize()])
+
 model1 = models.torchvision.faster_rcnn.model(backbone=models.torchvision.faster_rcnn.backbones.resnet18_fpn(pretrained=True), num_classes=len(class_map))
 state_dict = torch.load('fasterRCNN_resnet18_Raccoons.pth')
 model1.load_state_dict(state_dict)