Update app.py

app.py

@@ -20,6 +20,7 @@ backbone = model_type.backbones.resnet50_fpn
 model = model_type.model(backbone=backbone(pretrained=True), num_classes=len(parser.class_map), **extra_args) 
 
 import models
+model = model.pth
 
 
 # Transforms