Spaces:

Alesteba
/

AP_02

Sleeping

Alesteba commited on Apr 18, 2023

Commit

70fc88e

1 Parent(s): 4129c4b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ size = 384
 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),
     img_size = size
 )

 model_type = models.ross.efficientdet
 model_2 = model_type.model(
+    backbone= model_type.backbones.tf_d0 (pretrained=True),
     num_classes=len(class_map),
     img_size = size
 )