Spaces:

allispaul
/

foodvision_mini

Runtime error

allispaul commited on Jul 20, 2023

Commit

08f8524

•

1 Parent(s): 125b55b

fix typo

Files changed (1) hide show

model.py CHANGED Viewed

@@ -6,7 +6,7 @@ import torchvision
 def create_effnetb2_model(num_classes: int = 3,
                           seed: int = 4,
-                         ) -> Tuple[nn.Module, torchvision.Compose]:
     """Create an EfficientNetB2 feature extractor model and transforms.
     Args:

 def create_effnetb2_model(num_classes: int = 3,
                           seed: int = 4,
+                         ) -> Tuple[nn.Module, torchvision.transforms.Compose]:
     """Create an EfficientNetB2 feature extractor model and transforms.
     Args: