Update app.py

app.py

@@ -12,8 +12,8 @@ class_map = ClassMap(['raccoon','banana'])
 size = 384
 
 model_2 = models.torchvision.retinanet.model(
-    backbone=models.torchvision.retinanet.backbones.wide_resnet101_2_fpn (pretrained=True),
-    num_classes=len(class_map)
+    backbone=models.torchvision.retinanet.backbones.resnet18_fpn (pretrained=True),
+    num_classes=len(class_map),
 )
 
 # load from model_repo: