Update app.py

app.py

@@ -17,10 +17,10 @@ extra_args = {}
 model_type = models.torchvision.retinanet
 backbone = model_type.backbones.resnet50_fpn
 # Instantiate the model
-model = model_type.model(backbone=backbone(pretrained=True), num_classes=len(parser.class_map), **extra_args) 
+
+#model = model_type.model(backbone=backbone(pretrained=True), num_classes=len(parser.class_map), **extra_args) 
 
 import models
-model = model.pth
 
 
 # Transforms
@@ -35,8 +35,11 @@ valid_ds = Dataset(valid_records, valid_tfms)
 train_dl = model_type.train_dl(train_ds, batch_size=8, num_workers=4, shuffle=True)
 valid_dl = model_type.valid_dl(valid_ds, batch_size=8, num_workers=4, shuffle=False)
 metrics = [COCOMetric(metric_type=COCOMetricType.bbox)]
+
 learn = model_type.fastai.learner(dls=[train_dl, valid_dl], model=model, metrics=metrics)
+
 learn = learn.load('model')
+
 import os
 for root, dirs, files in os.walk(r'sample_images/'):
     for filename in files: