Spaces:

vvd2003
/

Image-Captioning

Runtime error

vvd2003 commited on Apr 12, 2023

Commit

47fdd32

•

1 Parent(s): df79ab5

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -12,7 +12,7 @@ class CNN(nn.Module):
         super().__init__()
         # Load pretrained Efficientnet-B2 model
-        self.model = models.efficientnet_b2(weights=models.EfficientNet_B2_Weights)
         # Frozen all layer of model
         if not train_model:

         super().__init__()
         # Load pretrained Efficientnet-B2 model
+        self.model = models.efficientnet_b2(weights=None)
         # Frozen all layer of model
         if not train_model: