Spaces:

Alesteba
/

AP_02

Sleeping

Alesteba commited on Apr 17, 2023

Commit

f051121

1 Parent(s): 4fa56e7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import os
 # from icevision.all import *
 import torch
 model_2 = models.torchvision.retinanet.model(
     backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),

 # from icevision.all import *
 import torch
+from torchvision import *
 model_2 = models.torchvision.retinanet.model(
     backbone=models.torchvision.retinanet.backbones.resnext50_32x4d_fpn (pretrained=True),