Spaces:

Alesteba
/

AP_02

Build error

Alesteba commited on Apr 17, 2023

Commit

0ca578a

1 Parent(s): 2c487ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ 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)
 )

 size = 384
 model_2 = models.torchvision.retinanet.model(
+    backbone=models.torchvision.retinanet.backbones.resnet50_fpn (pretrained=True),
     num_classes=len(class_map)
 )