Spaces:

BigTimeCoderSean
/

FoodVision_big

Runtime error

BigTimeCoderSean commited on Jan 3, 2023

Commit

2deafea

1 Parent(s): 07ae9e6

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -4,7 +4,7 @@ import torchvision
 from torch import nn
-def create_effnetb2_model(num_classes:int=3,
                           seed:int=42):
     """Creates an EfficientNetB2 feature extractor model and transforms.

 from torch import nn
+def create_effnetb2(num_classes:int=3,
                           seed:int=42):
     """Creates an EfficientNetB2 feature extractor model and transforms.