Update app.py

app.py

@@ -11,11 +11,12 @@ class_map = ClassMap(['raccoon','banana'])
 
 size = 384
 
-model_type = models.mmdet.retinanet
+model_type = models.ross.efficientdet
 
 model_2 = model_type.model(
     backbone= model_type.backbones.swin_t_p4_w7_fpn_1x_coco (pretrained=True),
-    num_classes=len(class_map)
+    num_classes=len(class_map),
+    img_size = size
 )
 
 # load from model_repo: