Update model.py
Browse files
model.py
CHANGED
@@ -82,11 +82,6 @@ def image_transform(image):
|
|
82 |
return image
|
83 |
|
84 |
|
85 |
-
# Create EffNetB2 pretrained weights, transforms and model
|
86 |
-
weights = torchvision.models.EfficientNet_B2_Weights.DEFAULT
|
87 |
-
transforms = weights.transforms()
|
88 |
-
model = torchvision.models.efficientnet_b2(weights=weights)
|
89 |
-
|
90 |
#Defining a model
|
91 |
# weights=weights=torchvision.models.VGG19_Weights.IMAGENET1K_V1
|
92 |
# cnn = models.vgg19(weights=weights).features.eval()
|
|
|
82 |
return image
|
83 |
|
84 |
|
|
|
|
|
|
|
|
|
|
|
85 |
#Defining a model
|
86 |
# weights=weights=torchvision.models.VGG19_Weights.IMAGENET1K_V1
|
87 |
# cnn = models.vgg19(weights=weights).features.eval()
|