Update app.py

app.py

@@ -13,8 +13,7 @@ size = 384
 
 model_2 = models.torchvision.retinanet.model(
     backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),
-    num_classes=len(class_map),
-    img_size = size
+    num_classes=len(class_map)
 )
 
 # load from model_repo: