Update app.py

app.py

@@ -9,9 +9,12 @@ import PIL
 
 class_map = ClassMap(['raccoon','banana'])
 
+size = 384
+
 model_2 = models.torchvision.retinanet.model(
     backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),
-    num_classes=len(class_map)
+    num_classes=len(class_map),
+    img_size = size
 )
 
 # load from model_repo:
@@ -25,8 +28,6 @@ model_2.load_state_dict(state_dict)
 
 # use test img:
 
-size = 384
-
 infer_tfms = tfms.A.Adapter([*tfms.A.resize_and_pad(size),tfms.A.Normalize()])
 
 def predict(img):